bugfixes

Signed-off-by: Ali Boubezari <[email protected]>

Author: aboubezari

modelopt/onnx/autocast/precisionconverter.py

@@ -1083,7 +1083,7 @@ def _should_skip_low_precision_input_conversion(
 
         This is used for nodes that have inputs that MUST remain in FP32.
         """
-        match self.low_precision_type:
+        match self.low_precision_type.str_short:
             case "fp16":
                 skip_inputs_map = SKIP_LOW_PRECISION_MAPPING_FP16
             case "bf16":
@@ -1094,6 +1094,8 @@ def _should_skip_low_precision_input_conversion(
         if node.op_type in skip_inputs_map:
             # Figure out the index of the input in the node input
             inputs_lst = list(node.input)
+            if input_name not in inputs_lst:
+                raise ValueError(f"Input {input_name} not found in node {node.name}.")
             input_index = inputs_lst.index(input_name)
             # Check if we should skip this input for low precision conversion
             return input_index in skip_inputs_map[node.op_type]