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

added assert in test.

上级 f359b692
...@@ -541,6 +541,7 @@ def test_inplace0(): ...@@ -541,6 +541,7 @@ def test_inplace0():
if (gemm_inplace in [n.op for n in f.maker.env.nodes]): if (gemm_inplace in [n.op for n in f.maker.env.nodes]):
print pp(f.maker.env.outputs[0]) print pp(f.maker.env.outputs[0])
raise Failure('gemm_inplace in graph') raise Failure('gemm_inplace in graph')
assert _dot22 in [n.op for n in f.maker.env.nodes]
f = inplace_func([X,Y,Z,a,b, R, S, c], f = inplace_func([X,Y,Z,a,b, R, S, c],
[Z * (c*Z + a * T.dot(X,Y) + b * T.dot(R,S).T)], mode='FAST_RUN') [Z * (c*Z + a * T.dot(X,Y) + b * T.dot(R,S).T)], mode='FAST_RUN')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论