提交 dc636a5a authored 作者: Frederic Bastien's avatar Frederic Bastien

assert that a test was done on the gpu.

上级 45cad3a4
...@@ -773,6 +773,9 @@ def test_many_arg_elemwise(): ...@@ -773,6 +773,9 @@ def test_many_arg_elemwise():
f = theano.function( symb_args, op_to_test(*symb_args), mode = mode ) f = theano.function( symb_args, op_to_test(*symb_args), mode = mode )
#theano.printing.debugprint(f) #theano.printing.debugprint(f)
outputs.append( f( * args) ) outputs.append( f( * args) )
#assert that the test was done on the gpu.
if mode is mode_with_gpu:
assert any([isinstance(node.op, cuda.GpuElemwise) for node in f.maker.env.nodes])
results_gpu, results_cpu = outputs results_gpu, results_cpu = outputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论