diff --git a/stacks/pinot-noir/definitions_acc_cuda.yaml b/stacks/pinot-noir/definitions_acc_cuda.yaml index 23cc17db419c6f645317c1d6efe2d175e0967c8c..30bdd68fb0213b67952710d81f977de319faadc7 100644 --- a/stacks/pinot-noir/definitions_acc_cuda.yaml +++ b/stacks/pinot-noir/definitions_acc_cuda.yaml @@ -4,6 +4,7 @@ definitions: - cuda_serial_codes: - cuda - cudnn + - namd ^fftw ~mpi - nccl - cuda_mpi_codes: diff --git a/stacks/pinot-noir/packages_acc_cuda.yaml b/stacks/pinot-noir/packages_acc_cuda.yaml index 8bb04980cbd6b3ba1e8a18b2791832c7aee3c209..3c9c4b0bc14ffc99f12a9c9f098e14ba085b38e0 100644 --- a/stacks/pinot-noir/packages_acc_cuda.yaml +++ b/stacks/pinot-noir/packages_acc_cuda.yaml @@ -45,6 +45,9 @@ packages: libxc: require: ['+cuda', 'cuda_arch={cuda_arch}'] + namd: + require: ['+cuda', '+single_node_gpu', 'cuda_arch={cuda_arch}'] + nvshmem: require: - spec: '+cuda cuda_arch={cuda_arch} +nccl +ucx +mpi'