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

Convert some of the old GpuCAReduce code to the new interface.

上级 a8c03627
......@@ -66,6 +66,10 @@ class test_GpuCAReduceCPY(test_CAReduce):
self.with_linker(gof.CLinker(), op, dtype=dtype,
test_nan=True)
def test_infer_shape(self):
for dtype in self.dtypes:
test_CAReduce.test_infer_shape(self, dtype)
class test_GpuCAReduce(test_GpuCAReduceCPY):
dtypes = ["float32"]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论