WIP: Dockerfile

Author: hameerabbasi

Dockerfile.gpu

@@ -0,0 +1,7 @@
+FROM python:3.12-slim
+
+ADD . .
+
+RUN pip install ".[cuda-12]"
+
+CMD ["python"]

pyproject.toml

@@ -7,7 +7,7 @@ name = "sparse"
 dynamic = ["version"]
 description = "Sparse n-dimensional arrays for the PyData ecosystem"
 readme = "README.md"
-dependencies = ["numpy>=1.17", "numba>=0.49"]
+dependencies = ["numpy>=1.17", "numba>=0.49", "array_api_compat>=1.11"]
 maintainers = [{ name = "Hameer Abbasi", email = "[email protected]" }]
 requires-python = ">=3.10"
 license = { file = "LICENSE" }
@@ -51,6 +51,8 @@ tests = [
     "pre-commit",
     "pytest-codspeed",
 ]
+cuda-12 = ["cupy-cuda12x"]
+cuda-11 = ["cupy-cuda11x"]
 tox = ["sparse[tests]", "tox"]
 notebooks = ["sparse[tests]", "nbmake", "matplotlib"]
 all = ["sparse[docs,tox,notebooks,mlir]", "matrepr"]

sparse/numba_backend/_sparse_array.py

@@ -53,8 +53,11 @@ def __init__(self, shape, fill_value=None):
         if data is not None:
             import array_api_compat
 
-            self.fill_value = array_api_compat.array_namespace(data).asarray(self.fill_value)
+            xp = array_api_compat.array_namespace(data)
+        else:
+            xp = np
 
+        self.fill_value = xp.asarray(self.fill_value)
         self.device  # noqa: B018
 
     dtype = None