diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py index b41dee818d..8a5df8988e 100644 --- a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py +++ b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py @@ -125,8 +125,7 @@ aten.renorm, aten.renorm_, aten.rot90, - aten.rsub.Scalar, - aten.rsub.Tensor, + aten.rsub, aten.select_backward, aten.select_scatter, aten.sgn, @@ -175,6 +174,7 @@ aten.linalg_vector_norm, aten.full, aten.repeat, + aten.var_mean, } torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = { aten._softmax.default,