提交 bdee24d6 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #4725 from Sentient07/faster-careduce

Speeding up local_gpua_careduce
......@@ -621,7 +621,8 @@ class GpuCAReduceCuda(GpuKernelBase, HideC, CAReduceDtype):
try:
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:
return False
return True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论