From ea6a2013186ce7790e4ea1c61513a752592554b9 Mon Sep 17 00:00:00 2001 From: HolyWu Date: Sun, 10 Mar 2024 12:59:55 +0800 Subject: [PATCH] Remove ops from decompositions where converters exist --- .../dynamo/lowering/_decomposition_groups.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py index de791851db..9a38771fee 100644 --- 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,