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

small fix.

上级 e25e5b42
...@@ -937,6 +937,7 @@ class test_fusion(unittest.TestCase): ...@@ -937,6 +937,7 @@ class test_fusion(unittest.TestCase):
topo_ = [x for x in topo if not isinstance(x.op,tcn.basic_ops.GpuFromHost)] topo_ = [x for x in topo if not isinstance(x.op,tcn.basic_ops.GpuFromHost)]
gpu_ = [x for x in topo if isinstance(x.op,tcn.basic_ops.GpuFromHost)] gpu_ = [x for x in topo if isinstance(x.op,tcn.basic_ops.GpuFromHost)]
assert len(gpu_)==len(sym_inputs) assert len(gpu_)==len(sym_inputs)
else: topo_=topo
if assert_len_topo: if assert_len_topo:
assert(len(topo_)==nb_elemwise) assert(len(topo_)==nb_elemwise)
assert(out_dtype==out.dtype) assert(out_dtype==out.dtype)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论