Restore temporary changes from release

.github/workflows/python-package.yml

@@ -104,7 +104,6 @@ jobs:
           retention-days: 7
 
   build-shared-libs-rocm:
-    if: false # Temporarily disabled
     strategy:
       matrix:
         os: [ubuntu-22.04]
@@ -152,7 +151,7 @@ jobs:
     needs:
       - build-shared-libs
       - build-shared-libs-cuda
-      #- build-shared-libs-rocm
+      - build-shared-libs-rocm
     strategy:
       matrix:
         os: [ubuntu-22.04, ubuntu-22.04-arm, windows-latest, macos-latest]

README.md

@@ -73,9 +73,9 @@ bitsandbytes has the following minimum requirements for all platforms:
         CDNA: gfx90a, gfx942<br>
         RDNA: gfx1100
       </td>
-      <td>🚧</td>
-      <td>🚧</td>
-      <td>🚧</td>
+      <td>✅</td>
+      <td>〰️</td>
+      <td>✅</td>
     </tr>
     <tr>
       <td></td>
@@ -85,16 +85,16 @@ bitsandbytes has the following minimum requirements for all platforms:
         Arc A-Series (Alchemist)<br>
         Arc B-Series (Battlemage)
       </td>
-      <td>🚧</td>
-      <td>🚧</td>
+      <td>✅</td>
+      <td>✅</td>
       <td>🚧</td>
     </tr>
     <tr>
       <td></td>
       <td>🟪 Intel Gaudi <br><code>hpu</code></td>
       <td>Gaudi1, Gaudi2, Gaudi3</td>
-      <td>🚧</td>
-      <td>🚧</td>
+      <td>✅</td>
+      <td>〰️</td>
       <td>❌</td>
     </tr>
     <tr>
@@ -139,8 +139,8 @@ bitsandbytes has the following minimum requirements for all platforms:
         Arc A-Series (Alchemist) <br>
         Arc B-Series (Battlemage)
       </td>
-      <td>🚧</td>
-      <td>🚧</td>
+      <td>✅</td>
+      <td>✅</td>
       <td>🚧</td>
     </tr>
     <tr>

bitsandbytes/__init__.py

@@ -35,6 +35,16 @@ supported_torch_devices = {
 if torch.cuda.is_available():
     from .backends.cuda import ops as cuda_ops
 
+if hasattr(torch, "xpu") and torch.xpu.is_available():
+    from .backends.xpu import ops as xpu_ops
+
+if importlib.util.find_spec("habana_frameworks") and importlib.util.find_spec("habana_frameworks.torch"):
+    # In case not automatically imported
+    import habana_frameworks.torch
+
+    if hasattr(torch, "hpu") and torch.hpu.is_available():
+        from .backends.hpu import ops as hpu_ops
+
 
 def _import_backends():
     """

pyproject.toml

@@ -74,7 +74,6 @@ test = [
 package-data = { "*" = ["libbitsandbytes*.*"] }
 
 [tool.setuptools.packages.find]
-exclude = ["*backends.xpu", "*backends.hpu", "*backends.triton"]
 include = ["bitsandbytes*"]
 
 [tool.setuptools.dynamic]