fix

Author: xadupre

onnx_diagnostic/torch_export_patches/patches/_patch_transformers_qwen2_5.py

@@ -262,12 +262,14 @@ def qwen_version_selector(opset: int, *args: torch.Tensor) -> Tuple[str, torch.d
         itype = torch_dtype_to_onnx_dtype(dtype)
         if strategy is not None:
             return strategy, itype
-        if dtype == torch.float32:
+        if dtype == torch.float32 or itype == onnx.TensorProto.FLOAT:
             if opset >= 24:
                 return "LOOPA24", itype
             return "LOOPMHA", itype
-        if dtype == torch.float16:
-            if first_tensor.is_cuda:
+        if dtype == torch.float16 or itype == onnx.TensorProto.FLOAT16:
+            # first_tensor may be a SymbolicTensor (onnx).
+            # is_cuda is not available.
+            if hasattr(first_tensor, "is_cuda") and first_tensor.is_cuda:
                 return "PACKED", itype
             return "LOOPMHA", itype
         raise AssertionError(