chore(dev-deps): add pixi

.gitattributes

@@ -0,0 +1,5 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-2-Clause
+
+# SCM syntax highlighting & preventing 3-way merges
+pixi.lock merge=binary linguist-language=YAML linguist-generated=true

.gitignore

@@ -15,3 +15,7 @@ testing/*.ltoir
 testing/*.ltoir.o
 testing/*.o
 testing/*.ptx
+# pixi environments
+.pixi/*
+!.pixi/config.toml
+*.log

.spdx-ignore

@@ -32,3 +32,4 @@ numba_cuda/numba/cuda/include/*/cuda_*.hpp
 
 # GitHub configuration files
 .github/CODEOWNERS
+pixi.lock

ci/coverage_report.sh

@@ -10,11 +10,12 @@ rapids-logger "Install wheel with test dependencies and coverage tools"
 package=$(realpath wheel/numba_cuda*.whl)
 echo "Package path: ${package}"
 python -m pip install \
-    "${package}[test]" \
+    "${package}" \
     "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*" \
     "cuda-core==0.3.*" \
     pytest-cov \
-    coverage
+    coverage \
+    --group test
 
 rapids-logger "Build test binaries"
 export NUMBA_CUDA_TEST_BIN_DIR=`pwd`/testing

ci/test_thirdparty.sh

@@ -20,10 +20,12 @@ rapids-logger "Install wheel with test dependencies"
 package=$(realpath wheel/numba_cuda*.whl)
 echo "Package path: ${package}"
 python -m pip install \
-    "${package}[test]" \
+    "${package}" \
     "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*" \
     "cuda-core==0.3.*" \
     "nvidia-nvjitlink-cu12" \
+    --group test
+
 
 
 rapids-logger "Shallow clone cuDF repository"

ci/test_wheel.ps1

@@ -31,7 +31,7 @@ $CUDA_VER_MAJOR = ($env:CUDA_VER -split '\.')[0] -join '.'
 rapids-logger "Install wheel with test dependencies"
 $package = Resolve-Path wheel\numba_cuda*.whl | Select-Object -ExpandProperty Path
 echo "Package path: $package"
-python -m pip install "${package}[cu${CUDA_VER_MAJOR},test-cu${CUDA_VER_MAJOR}]"
+python -m pip install "${package}[cu${CUDA_VER_MAJOR}]" --group "test-cu${CUDA_VER_MAJOR}"
 python -m pip install "llvmlite<0.45" "numba==0.61.*" # WAR for https://github.com/numba/llvmlite/issues/1297
 
 

ci/test_wheel.sh

@@ -11,9 +11,11 @@ package=$(realpath wheel/numba_cuda*.whl)
 echo "Package path: ${package}"
 
 DEPENDENCIES=(
-    "${package}[test]"
+    "${package}"
     "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*"
     "cuda-core==0.3.*"
+    "--group"
+    "test"
 )
 
 # Constrain oldest supported dependencies for testing

ci/test_wheel_ctypes_binding.sh

@@ -10,8 +10,9 @@ rapids-logger "Install wheel with testing dependencies"
 package=$(realpath wheel/numba_cuda*.whl)
 echo "Package path: $package"
 python -m pip install \
-    "${package}[test]" \
+    "${package}" \
     cuda-python \
+    --group test
 
 # FIXME: Find a way to build the tests that does not depend on the CUDA Python bindings
 #rapids-logger "Build tests"

ci/test_wheel_deps_wheels.sh

@@ -11,7 +11,7 @@ rapids-logger "Install wheel with test dependencies"
 package=$(realpath wheel/numba_cuda*.whl)
 echo "Package path: ${package}"
 # TODO: control minor version pinning to honor TEST_MATRIX once the cuda-toolkit metapackage is up
-python -m pip install "${package}[cu${CUDA_VER_MAJOR},test-cu${CUDA_VER_MAJOR}]"
+python -m pip install "${package}[cu${CUDA_VER_MAJOR}]" --group "test-cu${CUDA_VER_MAJOR}"
 
 rapids-logger "Build test binaries"
 

pyproject.toml

@@ -46,6 +46,7 @@ cu13 = [
     "nvidia-cuda-cccl==13.*",
 ]
 
+[dependency-groups]
 test = [
     "pre-commit",
     "psutil",
@@ -56,14 +57,12 @@ test = [
     "ml_dtypes",
 ]
 test-cu12 = [
-    "numba-cuda[cu12]",
-    "numba-cuda[test]",
     "nvidia-curand-cu12",
+    { include-group = "test" }
 ]
 test-cu13 = [
-    "numba-cuda[cu13]",
-    "numba-cuda[test]",
     "nvidia-curand==10.4.*",
+    { include-group = "test" }
 ]
 
 [project.urls]
@@ -146,3 +145,18 @@ exclude = [
 "numba_cuda/numba/cuda/tests/doc_examples/test_cg.py" = ["E501"]
 "numba_cuda/numba/cuda/tests/doc_examples/test_matmul.py" = ["E501"]
 "numba_cuda/numba/tests/doc_examples/test_interval_example.py" = ["E501"]
+
+[tool.pixi.workspace]
+channels = ["conda-forge"]
+platforms = ["linux-64", "linux-aarch64", "win-64"]
+
+[tool.pixi.pypi-dependencies]
+numba-cuda = { path = ".", editable = true }
+
+[tool.pixi.environments]
+cu12 = { features = ["cu12", "test"], solve-group = "cu12" }
+cu13 = { features = ["cu13", "test"], solve-group = "cu13" }
+test = { features = ["test"], solve-group = "test" }
+
+[tool.pixi.tasks]
+test = { cmd = ["pytest", "$PIXI_PROJECT_ROOT/numba_cuda/numba/cuda/tests"] }