update

huggingface · Nov 24, 2023 · 1551292 · 1551292
1 parent 0fe7545
commit 1551292
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/optimum/onnxruntime/utils.py b/optimum/onnxruntime/utils.py
@@ -197,10 +197,10 @@ def get_provider_for_device(device: torch.device) -> str:
     Gets the ONNX Runtime provider associated with the PyTorch device (CPU/CUDA).
     """
     if device.type.lower() == "cuda":
-        if "CUDAExecutionProvider" in ort.get_available_providers():
-            return "CUDAExecutionProvider"
-        else:
+        if "ROCMExecutionProvider" in ort.get_available_providers():
             return "ROCMExecutionProvider"
+        else:
+            return "CUDAExecutionProvider"
     return "CPUExecutionProvider"