提交 60a88f29 authored 作者: Adam Becker's avatar Adam Becker

move topk .c* files into subdir

上级 901a7a0b
...@@ -62,12 +62,12 @@ class GpuTopKOp(GpuKernelBase, TopKOp): ...@@ -62,12 +62,12 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
kernel_src = {} kernel_src = {}
for kname in knames: for kname in knames:
with open(os.path.join( with open(os.path.join(
os.path.dirname(__file__), kname + kernel_ext os.path.dirname(__file__), 'c_code', kname + kernel_ext
), 'r') as f: ), 'r') as f:
kernel_src[kname] = f.read() kernel_src[kname] = f.read()
with open(os.path.join( with open(os.path.join(
os.path.dirname(__file__), 'k_topk_common' + common_ext os.path.dirname(__file__), 'c_code', 'k_topk_common' + common_ext
), 'r') as f: ), 'r') as f:
common_src = f.read() common_src = f.read()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论