We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 21eeb00 + cf8fe0a commit a73c571Copy full SHA for a73c571
.github/workflows/setup_env.sh
@@ -150,9 +150,10 @@ if [ "${device}" = "gpu_nvidia" ]; then
150
151
elif [ "${device}" = "gpu_amd" ]; then
152
print "======================================== Load ROCm"
153
- export ROCM_PATH=/opt/rocm
+ export MODULEPATH=$MODULEPATH:/apps/rocm/Modulefiles
154
+ quiet module load rocm/6.3.2
155
# Some hip utilities require /usr/sbin/lsmod
- export PATH=${PATH}:${ROCM_PATH}/bin:/usr/sbin
156
+ export PATH=${PATH}:/usr/sbin
157
export gpu_backend=hip
158
quiet which hipcc
159
hipcc --version
0 commit comments