提交 dbe33b01 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix crash in DebugMode and don't use the CPU perform on a GPU op that didn't implement it.

上级 41509b28
......@@ -48,6 +48,9 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
return_values=return_values,
return_indices=return_indices)
def perform(self, node, inputs, output_storage, params):
raise NotImplementedError()
def c_headers(self):
return ['gpuarray_api.h', 'gpuarray_helper.h', 'numpy_compat.h']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论