nvidia-nvshmem-cu12
cuda-python<=12.9,>=12.0
cuda.core>=0.5.0
cuda.pathfinder>=1.2.3
numpy
Cython>=0.29.24

[cute]
nvidia-cutlass-dsl<5.0,>=4.5.2

[dev]
pytest>=6.0
pytest-mpi
black
isort

[mlir]
numba-cuda-mlir[cu12]>=0.4.2

[numba]
numba<0.66
numba-cuda[cu12]>=0.28.0
