Remove ops from decompositions where converters exist

pytorch · Mar 10, 2024 · ea6a201 · ea6a201
1 parent ad74a73
commit ea6a201
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -35,7 +35,6 @@
     aten.diagonal_backward,
     aten.dot,
     aten.elu_backward,
-    aten._embedding_bag,
     aten.embedding_dense_backward,
     aten._euclidean_dist.default,
     aten.expand_as,
@@ -46,10 +45,8 @@
     aten.gelu,
     aten.gelu_backward,
     aten.glu_backward,
-    aten.grid_sampler_2d,
     aten.hardshrink,
     aten.hardshrink_backward,
-    aten.hardsigmoid,
     aten.hardsigmoid_backward,
     aten.hardswish,
     aten.hardswish_,
@@ -71,7 +68,6 @@
     aten.isneginf,
     aten.isposinf,
     aten.l1_loss,
-    aten.leaky_relu,
     aten.leaky_relu_,
     aten.leaky_relu_backward,
     aten.lerp,
@@ -135,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,
@@ -157,8 +152,6 @@
     aten.unfold_backward,
     aten.unfold_copy,
     aten._unsafe_index,
-    aten.upsample_bilinear2d,
-    aten.upsample_bilinear2d.vec,
     aten.upsample_nearest2d_backward,
     aten.var,
     aten.var_mean,
@@ -173,7 +166,6 @@
     aten.linalg_vector_norm,
     aten.full,
     aten.repeat,
-    aten.var_mean,
 }
 torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = {
     aten._softmax.default,