提交 36a857de authored 作者: sentient07's avatar sentient07

Speeding up local_gpua_careduce

上级 195f9b1d
...@@ -621,7 +621,8 @@ class GpuCAReduceCuda(GpuKernelBase, HideC, CAReduceDtype): ...@@ -621,7 +621,8 @@ class GpuCAReduceCuda(GpuKernelBase, HideC, CAReduceDtype):
try: try:
self.c_code(node, name, inp, out, sub) self.c_code(node, name, inp, out, sub)
self.c_support_code_apply(node, name) if not self.gpu_kernels(node, name):
return False
except NotImplementedError: except NotImplementedError:
return False return False
return True return True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论