Skip to content

Commit 8eded6d

Browse files
committed
build: Add multicore-cuda option for NAMD test builds
1 parent 4ae0a39 commit 8eded6d

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

devel-tools/compile-namd.sh

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,27 @@ EOF
7777
cmd+=(--charm-arch multicore-linux-${charm_arch_string})
7878
fi
7979

80+
if [ "${label}" = "multicore-cuda" ] ; then
81+
cmd+=(--charm-arch multicore-linux-${charm_arch_string})
82+
if ! hash nvcc >& /dev/null ; then
83+
if [ -x /usr/local/cuda/bin/nvcc ] ; then
84+
export PATH=/usr/local/cuda/bin:${PATH}
85+
fi
86+
fi
87+
if hash nvcc >& /dev/null ; then
88+
CUDA_HOME=$(which nvcc)
89+
export CUDA_HOME=${CUDA_HOME%/bin/nvcc}
90+
cmd+=(--with-cuda --cuda-prefix ${CUDA_HOME})
91+
if [ -f src/SequencerCUDA.C ] ; then
92+
# NAMD >= 3.0
93+
cmd+=(--with-single-node-cuda)
94+
fi
95+
else
96+
echo "Error: Missing CUDA compiler." >& 2
97+
return 1
98+
fi
99+
fi
100+
80101
if [ "${label}" = "mpi" ] ; then
81102
cmd+=(--charm-arch mpi-linux-${charm_arch_string}-smp)
82103
fi

0 commit comments

Comments
 (0)