提交 7399bc43 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add test.

上级 af3d84bd
......@@ -206,6 +206,14 @@ class TestSpecifyShape(test_basic.TestSpecifyShape):
input_type = GpuArrayType
class test_gpu_ifelse(theano.tests.test_ifelse.test_ifelse):
mode = mode_with_gpu
@staticmethod
def cast_output(v):
return basic_ops.as_gpuarray_variable(v, test_ctx)
shared = staticmethod(gpuarray_shared_constructor)
def test_print_op():
""" Test that print ops don't block gpu optimization"""
b = tensor.fmatrix()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论