diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py index bfb3d9545c..1fca374dcb 100644 --- a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py +++ b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py @@ -48,7 +48,6 @@ aten.glu_backward, aten.hardshrink, aten.hardshrink_backward, - aten.hardsigmoid, aten.hardsigmoid_backward, aten.hardswish, aten.hardswish_, @@ -66,11 +65,9 @@ aten.index_copy_, aten.index_fill, aten.index_fill_, - aten.index_select, aten.isneginf, aten.isposinf, aten.l1_loss, - aten.leaky_relu, aten.leaky_relu_, aten.leaky_relu_backward, aten.lerp, @@ -134,7 +131,6 @@ aten.soft_margin_loss_backward, aten._softmax.out, aten._softmax_backward_data, - aten.softplus, aten.softplus_backward, aten.softshrink, aten.softshrink_backward, @@ -172,7 +168,6 @@ aten.linalg_vector_norm, aten.full, aten.repeat, - aten.var_mean, } torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = { aten._softmax.default,