add npu to supported devices

bitsandbytes-foundation · Dec 6, 2024 · 51304ff · 51304ff
1 parent fe9d837
commit 51304ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py
@@ -25,6 +25,7 @@
 features = {"multi_backend"}
 supported_torch_devices = {
     "cuda",  # includes ROCm
+    "npu",  # Ascend NPU
     "xpu",  # Intel GPU
     "cpu",
 }