Re-activate GpuTopK

theano/gpuarray/sort.py

@@ -326,9 +326,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

theano/gpuarray/tests/test_sort.py

@@ -4,13 +4,9 @@ import theano
 import theano.tensor.tests.test_sort
 from .config import mode_with_gpu
 from ..sort import GpuTopKOp
-from nose.plugins.skip import SkipTest
 
 
 class Test_GpuTopK(theano.tensor.tests.test_sort.Test_TopK):
     mode = mode_with_gpu
     dtype = 'float32'
     op_class = GpuTopKOp
-
-    def setUp(self):
-        raise SkipTest("GpuTopK is currently disabled")