DiHydrogen: Specify required NVSHMEM variants (#26384)

2021-10-03 05:30:27 -07:00
parent f6ce95bfcb
commit 2de116d285
1 changed files with 2 additions and 1 deletions
--- a/var/spack/repos/builtin/packages/dihydrogen/package.py
+++ b/var/spack/repos/builtin/packages/dihydrogen/package.py
@@ -122,7 +122,8 @@ class Dihydrogen(CMakePackage, CudaPackage, ROCmPackage):

    depends_on('llvm-openmp', when='%apple-clang +openmp')

-    depends_on('nvshmem', when='+nvshmem')
+    # TODO: Debug linker errors when NVSHMEM is built with UCX
+    depends_on('nvshmem +nccl~ucx', when='+nvshmem')

    # Idenfity versions of cuda_arch that are too old
    # from lib/spack/spack/build_systems/cuda.py