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

Disable GpuTopK for now.

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