提交 b172a19b authored 作者: Frederic's avatar Frederic

Correctly test gemv init behavior when device=gpu and don't profile it.

上级 c7216204
...@@ -730,7 +730,9 @@ def check_force_gemv_init(): ...@@ -730,7 +730,9 @@ def check_force_gemv_init():
f = theano.function( f = theano.function(
[aa, yy, xx], [aa, yy, xx],
gemv_no_inplace(aa, 1., xx, yy, 0.), gemv_no_inplace(aa, 1., xx, yy, 0.),
theano.compile.Mode(optimizer='fast_compile') theano.compile.Mode(optimizer='fast_compile').excluding('gpu',
'gpuarray'),
profile=False
) )
finally: finally:
theano.config.compute_test_value = tv theano.config.compute_test_value = tv
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论