In pyproject.toml, it might be worth it (after numba-cuda-mlir is stable) to eventually add:
[tool.pixi.feature.cuda]
platforms = ["linux-64", "linux-aarch64"]
[tool.pixi.feature.cuda.dependencies]
cuda-nvcc = "*"
numba-cuda-mlir = "*"
[tool.pixi.environments]
cuda = ["cuda"]
And then, this can be activated via:
or
pixi run -e cuda tests gpu
In
pyproject.toml, it might be worth it (after numba-cuda-mlir is stable) to eventually add:And then, this can be activated via:
or