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

Consider all CAReduce subclasses for optimization

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