test

pytorch · Sep 22, 2024 · 4a26ee8 · 4a26ee8
1 parent ca7dec2
commit 4a26ee8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/torch_tensorrt/dynamo/lowering/_decompositions.py b/py/torch_tensorrt/dynamo/lowering/_decompositions.py
@@ -395,4 +395,4 @@ def get_decompositions(
         return {**CORE_ATEN_DECOMPOSITIONS_FILTERED, **TORCH_TRT_DECOMPOSITIONS}
     else:
         # changes made here due to torch2.6 changes https://github.com/pytorch/pytorch/pull/135080
-        return {**_decomp_table_to_post_autograd_aten(), **TORCH_TRT_DECOMPOSITIONS}
+        return {**_decomp_table_to_post_autograd_aten(), **ENABLED_TORCH_DECOMPOSITIONS, **TORCH_TRT_DECOMPOSITIONS}