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

use smaller shape to prevent DebugMode to return error sometimes due to…

use smaller shape to prevent DebugMode to return error sometimes due to different rounding on CPU and GPU with the Sum op.
上级 5626a476
......@@ -121,7 +121,7 @@ def test_memory_lazy():
We need to test it as the ifelse op with the [c]vm create op not
executed in the graph. This mess with [c]vm gc implementation.
"""
shapes = (200, 100)
shapes = (50, 100)
# more_alloc1 and more_alloc2 is not the same for both dtype.
# when dtype is float32, the computation is done on the gpu.
# This insert constant on the gpu during compilation
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论