提交 f6e09009 authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Consider all CAReduce subclasses for optimization

上级 984404ff
......@@ -6032,7 +6032,7 @@ ALL_REDUCE = [
T.elemwise.Sum,
T.elemwise.Prod,
T.elemwise.ProdWithoutZeros,
]
] + T.elemwise.CAReduce.__subclasses__()
@register_canonicalize
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论