From ab3bfaf1c5203e369a04c806f207927896e43a12 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Sun, 8 Jun 2025 08:33:35 -0700 Subject: [PATCH 1/2] fix issue 3560 --- .../dynamo/lowering/passes/constant_folding.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py b/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py index 0feec63316..3199786044 100644 --- a/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py +++ b/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py @@ -1,3 +1,4 @@ +import importlib import logging from typing import Any @@ -101,6 +102,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # TODO: Update this function when quantization is added def is_impure(self, node: torch.fx.node.Node) -> bool: - if node.target in (torch.ops.tensorrt.quantize_op.default,): - return True + if importlib.util.find_spec("modelopt.torch.quantization"): + if node.target in (torch.ops.tensorrt.quantize_op.default,): + return True return False From 3bfbac83e424299d40c7f9539fc2a2a454f303ad Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Sun, 8 Jun 2025 08:48:08 -0700 Subject: [PATCH 2/2] test --- py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py | 2 +- 1 file changed, 1 insertion(+), 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 3199786044..10fddec5c0 100644 --- a/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py +++ b/py/torch_tensorrt/dynamo/lowering/passes/constant_folding.py @@ -102,7 +102,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # TODO: Update this function when quantization is added def is_impure(self, node: torch.fx.node.Node) -> bool: - if importlib.util.find_spec("modelopt.torch.quantization"): + if importlib.util.find_spec("modelopt"): if node.target in (torch.ops.tensorrt.quantize_op.default,): return True return False