upgrade torch from 2.5.0.dev to 2.6.0.dev in main branch by lanluo-nvidia · Pull Request #3165 · pytorch/TensorRT
--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/lowering/_decompositions.py 2024-09-22 17:33:28.426941+00:00 +++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/lowering/_decompositions.py 2024-09-22 17:33:50.978789+00:00 @@ -393,6 +393,10 @@ if decomp not in torch_disabled_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(), **ENABLED_TORCH_DECOMPOSITIONS, **TORCH_TRT_DECOMPOSITIONS} + return { + **_decomp_table_to_post_autograd_aten(), + **ENABLED_TORCH_DECOMPOSITIONS, + **TORCH_TRT_DECOMPOSITIONS, + }