提交 55de6257 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test following disabling of gpu inplace gemm.(They execute not inplace)

上级 dac6eadd
...@@ -80,7 +80,7 @@ def test_gemm(): ...@@ -80,7 +80,7 @@ def test_gemm():
c = tensor.fmatrix('c') c = tensor.fmatrix('c')
f = pfunc([b,c], [], updates=[(a, tensor.dot(a,b) + tensor.exp(c))], mode=mode_with_gpu) f = pfunc([b,c], [], updates=[(a, tensor.dot(a,b) + tensor.exp(c))], mode=mode_with_gpu)
assert any([node.op == tcn.blas.gpu_gemm_inplace for node in f.maker.env.toposort()]) assert any([node.op == tcn.blas.gpu_gemm_no_inplace for node in f.maker.env.toposort()])
a0 = a.get_value() * 1.0 a0 = a.get_value() * 1.0
print a0 print a0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论