提交 471e2313 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add subtypes of CAReduce since the match is type-exact.

上级 45ce2b57
...@@ -602,7 +602,13 @@ def local_gpu_gemm(node): ...@@ -602,7 +602,13 @@ def local_gpu_gemm(node):
@register_opt() @register_opt()
@local_optimizer([tensor.elemwise.CAReduce]) @local_optimizer([tensor.elemwise.CAReduce,
tensor.elemwise.All,
tensor.elemwise.Any,
tensor.elemwise.CAReduceDtype,
tensor.elemwise.Sum,
tensor.elemwise.Prod,
tensor.elemwise.ProdWithoutZeros])
def local_gpu_careduce(node): def local_gpu_careduce(node):
if isinstance(node.op, tensor.elemwise.CAReduce): if isinstance(node.op, tensor.elemwise.CAReduce):
scalar_op = node.op.scalar_op scalar_op = node.op.scalar_op
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论