提交 62ce362d authored 作者: Adam Becker's avatar Adam Becker

remove "k_" from .cu files

上级 7a5cba10
...@@ -107,7 +107,7 @@ class GpuTopKOp(GpuKernelBase, TopKOp): ...@@ -107,7 +107,7 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
# load and compile kernels # load and compile kernels
with open(os.path.join( with open(os.path.join(
os.path.dirname(__file__), 'c_code', 'k_topk_common' + common_ext os.path.dirname(__file__), 'c_code', 'topk_common' + common_ext
)) as f: )) as f:
common_src = f.read() common_src = f.read()
...@@ -128,14 +128,14 @@ class GpuTopKOp(GpuKernelBase, TopKOp): ...@@ -128,14 +128,14 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
subs['count_t'] = 'int' subs['count_t'] = 'int'
kernels.append( kernels.append(
build_kernel('k_topk_dense' + kernel_ext, 'k_topk_dense', subs)) build_kernel('topk_dense' + kernel_ext, 'k_topk_dense', subs))
subs['kname'] = 'k_topk_dense_large' subs['kname'] = 'topk_dense_large'
kernels.append( kernels.append(
build_kernel('k_topk_dense_large' + kernel_ext, 'k_topk_dense_large', subs)) build_kernel('topk_dense_large' + kernel_ext, 'k_topk_dense_large', subs))
subs['count_t'] = 'long long' subs['count_t'] = 'long long'
subs['kname'] = 'k_topk_dense_xlarge' subs['kname'] = 'topk_dense_xlarge'
kernels.append( kernels.append(
build_kernel('k_topk_dense_large' + kernel_ext, 'k_topk_dense_xlarge', subs)) build_kernel('topk_dense_large' + kernel_ext, 'k_topk_dense_xlarge', subs))
return kernels return kernels
def c_code(self, node, nodename, inps, outs, sub): def c_code(self, node, nodename, inps, outs, sub):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论