Skip to content

Commit 178f4f3

Browse files
authored
Enabling var_mean decomposition (#2273)
1 parent e49ef6d commit 178f4f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,7 @@
125125
aten.renorm,
126126
aten.renorm_,
127127
aten.rot90,
128-
aten.rsub.Scalar,
129-
aten.rsub.Tensor,
128+
aten.rsub,
130129
aten.select_backward,
131130
aten.select_scatter,
132131
aten.sgn,
@@ -175,6 +174,7 @@
175174
aten.linalg_vector_norm,
176175
aten.full,
177176
aten.repeat,
177+
aten.var_mean,
178178
}
179179
torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = {
180180
aten._softmax.default,

0 commit comments

Comments
 (0)