提交 2fb56225 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: notoraptor

Disable GpuTopK for now.

上级 dcd37993
......@@ -300,9 +300,9 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
return node.inputs[0].type.context
@register_opt('fast_compile')
#@register_opt('fast_compile')
@op_lifter([TopKOp], cuda_only=True)
@register_opt2([TopKOp], 'fast_compile')
#@register_opt2([TopKOp], 'fast_compile')
def local_gpua_topkop(op, ctx_name, inputs, outputs):
axis = op.axis
rv = op.return_values
......
......@@ -4,6 +4,7 @@ import theano
import theano.tensor.tests.test_sort
from .config import mode_with_gpu, test_ctx_name
from ..sort import GpuTopKOp
from nose.plugins.skip import SkipTest
class Test_GpuTopK(theano.tensor.tests.test_sort.Test_TopK):
......@@ -11,4 +12,5 @@ class Test_GpuTopK(theano.tensor.tests.test_sort.Test_TopK):
dtype = 'float32'
op_class = GpuTopKOp
\ No newline at end of file
def setUp(self):
raise SkipTest("GpuTopK is currently disabled")
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论