Skip to content

Commit 6b7fdbb

Browse files
cherry pick 3660 to release/2.8 (#3661)
1 parent 1b2d668 commit 6b7fdbb

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.github/workflows/build-test-windows.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ jobs:
230230
cd dynamo
231231
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
232232
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
233-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
233+
../../../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
234234
popd
235235
236236
tests-py-dynamo-core:

py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,14 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
103103
self.quantization_ops: Set[torch._ops.OpOverload] = set()
104104
try:
105105
# modelopt import ensures torch.ops.tensorrt.quantize_op.default is registered
106-
import modelopt.torch.quantization as mtq
106+
import modelopt.torch.quantization as mtq # noqa: F401
107107

108108
assert torch.ops.tensorrt.quantize_op.default
109+
assert torch.ops.tensorrt.dynamic_block_quantize_op.default
109110
self.quantization_ops.add(torch.ops.tensorrt.quantize_op.default)
111+
self.quantization_ops.add(
112+
torch.ops.tensorrt.dynamic_block_quantize_op.default
113+
)
110114
except Exception as e:
111115
pass
112116

0 commit comments

Comments
 (0)