提交 0d867ade authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add missing op in list.

上级 06be9e5b
......@@ -130,9 +130,13 @@ def test_reduce():
ops = [type(node.op) for node in topo]
if kind == b'opencl' and method in ["max", "min"]:
assert not(GpuCAReduceCuda in ops or GpuCAReduceCPY in ops)
assert not(GpuCAReduceCuda in ops or
GpuCAReduceCPY in ops or
GpuDnnReduction in ops)
else:
assert GpuCAReduceCuda in ops or GpuCAReduceCPY in ops
assert (GpuCAReduceCuda in ops or
GpuCAReduceCPY in ops or
GpuDnnReduction in ops)
def test_local_gpualloc_memset_0():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论