From b27d0515276f2463e29eb087e64bc1ca72d81078 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Mon, 7 Jul 2025 11:02:13 -0700 Subject: [PATCH 1/2] fix compiler cl not found error in windows --- .github/workflows/build-test-windows.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build-test-windows.yml b/.github/workflows/build-test-windows.yml index e532efcced..e4fefbcef2 100644 --- a/.github/workflows/build-test-windows.yml +++ b/.github/workflows/build-test-windows.yml @@ -230,7 +230,7 @@ jobs: cd dynamo python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/ python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_complete_be_e2e_test_results.xml --ir torch_compile models/test_models.py - python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py + ../../../packaging/vc_env_helper.bat python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py popd tests-py-dynamo-core: From b45223a1621153a3eacf2e8cf10e9d7764323425 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Thu, 10 Jul 2025 08:54:41 -0700 Subject: [PATCH 2/2] constant folding --- .../dynamo/lowering/passes/constant_folding.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py b/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py index 928b7284fe..5ba84b09b0 100644 --- a/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py +++ b/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py @@ -103,10 +103,14 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: self.quantization_ops: Set[torch._ops.OpOverload] = set() try: # modelopt import ensures torch.ops.tensorrt.quantize_op.default is registered - import modelopt.torch.quantization as mtq + import modelopt.torch.quantization as mtq # noqa: F401 assert torch.ops.tensorrt.quantize_op.default + assert torch.ops.tensorrt.dynamic_block_quantize_op.default self.quantization_ops.add(torch.ops.tensorrt.quantize_op.default) + self.quantization_ops.add( + torch.ops.tensorrt.dynamic_block_quantize_op.default + ) except Exception as e: pass