Remove ops from decompositions where converters exist

pytorch · May 3, 2024 · a6e3e4b · a6e3e4b
1 parent 4dc9acf
commit a6e3e4b
Showing 1 changed file with 0 additions and 5 deletions.
diff --git 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,