From 1ad2aeba6e7255abe65f1ba1ffc856ea20cc6a63 Mon Sep 17 00:00:00 2001 From: apbose Date: Mon, 28 Aug 2023 12:33:21 -0700 Subject: [PATCH 1/2] Enabling var_mean decomposition --- py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py | 1 + 1 file changed, 1 insertion(+) diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py index b41dee818d..d3e090a831 100644 --- a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py +++ b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py @@ -175,6 +175,7 @@ aten.linalg_vector_norm, aten.full, aten.repeat, + aten.var_mean, } torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = { aten._softmax.default, From e349c36a0d1257534096b6519299399b36afe003 Mon Sep 17 00:00:00 2001 From: apbose Date: Mon, 28 Aug 2023 15:14:23 -0700 Subject: [PATCH 2/2] Enabling aten.rsub decomposition --- py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py index d3e090a831..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,