提交 50e62467 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix the test.

上级 dbf22bbe
...@@ -214,6 +214,9 @@ class test_gpu_ifelse(test_ifelse.test_ifelse): ...@@ -214,6 +214,9 @@ class test_gpu_ifelse(test_ifelse.test_ifelse):
return basic_ops.as_gpuarray_variable(v, test_ctx_name) return basic_ops.as_gpuarray_variable(v, test_ctx_name)
shared = staticmethod(gpuarray_shared_constructor) shared = staticmethod(gpuarray_shared_constructor)
def get_ifelse(self, n):
return theano.ifelse.IfElse(n, gpu=True, as_view=True)
def test_print_op(): def test_print_op():
""" Test that print ops don't block gpu optimization""" """ Test that print ops don't block gpu optimization"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论