From 52062dd798edd358e1e235963d1bc74d3b8c5d5a Mon Sep 17 00:00:00 2001
From: PiotrBLL <piotr.sobieszczyk@bluelabellabs.com>
Date: Thu, 31 Oct 2024 13:56:55 +0100
Subject: [PATCH] Add hpu_utils

---
 whisper/hpu_utils.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 whisper/hpu_utils.py

diff --git a/whisper/hpu_utils.py b/whisper/hpu_utils.py
new file mode 100644
index 0000000..7a66aa5
--- /dev/null
+++ b/whisper/hpu_utils.py
@@ -0,0 +1,15 @@
+import warnings
+
+import torch
+
+
+def load_default_hpu() -> str:
+    """
+    Load HPU if available, otherwise use CUDA or CPU.
+    """
+
+    if not torch.hpu.is_available():
+        warnings.warn("HPU is not available; trying to use CUDA instead.")
+        return "cuda" if torch.cuda.is_available() else "cpu"
+
+    return "hpu"