diff --git a/etc/spack/defaults/packages.yaml b/etc/spack/defaults/packages.yaml index b9fdd4b3dba..d52a035b0c3 100644 --- a/etc/spack/defaults/packages.yaml +++ b/etc/spack/defaults/packages.yaml @@ -21,6 +21,7 @@ packages: armci: [armcimpi] blas: [openblas, amdblis] c: [gcc] + cub: [cub] cxx: [gcc] D: [ldc] daal: [intel-oneapi-daal] diff --git a/var/spack/repos/builtin/packages/cuda/package.py b/var/spack/repos/builtin/packages/cuda/package.py index ea7b314418d..a08f68db9bf 100644 --- a/var/spack/repos/builtin/packages/cuda/package.py +++ b/var/spack/repos/builtin/packages/cuda/package.py @@ -702,6 +702,47 @@ class Cuda(Package): provides("opencl@:1.2", when="@7:") provides("opencl@:1.1", when="@:6") + for cuda_ver, cub_ver in [ + ["12.6.3", "2.5.0"], + ["12.6.2", "2.5.0"], + ["12.6.1", "2.5.0"], + ["12.6.0", "2.5.0"], + ["12.5.1", "2.4.0"], + ["12.5.0", "2.4.0"], + ["12.4.1", "2.3.1"], + ["12.4.0", "2.3.1"], + ["12.3.2", "2.2.0"], + ["12.3.1", "2.2.0"], + ["12.3.0", "2.2.0"], + ["12.2.2", "2.1.0"], + ["12.2.1", "2.1.0"], + ["12.2.0", "2.0.1"], + ["12.1.1", "2.0.1"], + ["12.1.0", "2.0.1"], + ["12.0.1", "2.0.1"], + ["12.0.0", "2.0.1"], + ["11.8.0", "1.15.1"], + ["11.7.1", "1.15.0"], + ["11.7.0", "1.15.0"], + ["11.6.2", "1.15.0"], + ["11.6.1", "1.15.0"], + ["11.6.0", "1.15.0"], + ["11.5.2", "1.13.1"], + ["11.5.1", "1.13.1"], + ["11.5.0", "1.13.1"], + ["11.4.4", "1.12.1"], + ["11.4.3", "1.12.1"], + ["11.4.2", "1.12.1"], + ["11.4.1", "1.12.1"], + ["11.4.0", "1.12.1"], + ["11.3.1", "1.11.0"], + ["11.3.0", "1.11.0"], + ["11.2.2", "1.10.0"], + ["11.2.1", "1.10.0"], + ["11.2.0", "1.10.0"], + ]: + provides(f"cub@{cub_ver}", when=f"@{cuda_ver}") + @classmethod def determine_version(cls, exe): output = Executable(exe)("--version", output=str, error=str)