提交 8073ca18 authored 作者: Faruk Ahmed's avatar Faruk Ahmed

include mode in function

上级 8c22815f
......@@ -471,7 +471,7 @@ def test_many_arg_elemwise():
for mode in [mode_with_gpu, mode_without_gpu]:
# test the optimization local_gpua_elemwise
output = op_to_test(*symb_args)
f = theano.function(symb_args, output)
f = theano.function(symb_args, output, mode=mode)
outputs.append(f(*args))
# assert that the test was done on the gpu.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论