Moved unsupported op detection out of constant

Signed-off-by: gcunhase <[email protected]>

Author: gcunhase

modelopt/onnx/autocast/precisionconverter.py

@@ -138,8 +138,8 @@ def __init__(
         self.min_opset = min_opset
         self.max_ir_version = max_ir_version
         self.trt_plugins = trt_plugins
-        OP_TYPES_NOT_SUPPORTED_IN_LOW_PRECISION.extend(
-            utils.get_ops_without_low_precision_support(
+        self.op_types_not_supported_in_low_precision = OP_TYPES_NOT_SUPPORTED_IN_LOW_PRECISION + (
+            utils.get_op_types_not_supported_in_low_precision(
                 self.model, self.low_precision_type.str_full, self.min_opset
             )
         )
@@ -451,7 +451,7 @@ def _filter_unsupported_op_types(
         # precision so we need to set Resize and Upsample to high precision
         for node in self.model.graph.node:
             if (
-                node.op_type in OP_TYPES_NOT_SUPPORTED_IN_LOW_PRECISION
+                node.op_type in self.op_types_not_supported_in_low_precision
                 and node.name in low_precision_nodes
             ):
                 low_precision_nodes.remove(node.name)

modelopt/onnx/autocast/utils.py

@@ -120,20 +120,20 @@ def get_cast_to_type(cast_node: onnx.NodeProto) -> int:
     raise ValueError("Cast node does not have 'to' attribute")
 
 
-def get_ops_without_low_precision_support(
+def get_op_types_not_supported_in_low_precision(
     model: onnx.ModelProto,
     low_precision_type: str,
     min_opset: int,
 ) -> list[str]:
-    """Get a list of ops without low precision support for the current opset version.
+    """Get a list of ops not supported in low precision for the current opset version.
 
     Args:
         model: ONNX model.
         low_precision_type: Target precision to reduce to ('float16' or 'bfloat16').
         min_opset: Minimum opset version.
 
     Returns:
-        ops_without_support: List of ops without low precision support for the current opset version.
+        ops_without_support: List of ops not supported in low precision for the current opset version.
     """
     # Obtain the current model's opset version
     ai_onnx_domain = [