Don't use hardcoded rocm path

This commit is contained in:
Tom Lin 2021-05-17 20:00:00 +01:00
parent cc16547e4d
commit 45ebd09ef2

View File

@ -240,8 +240,7 @@ setup_rocm() {
echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/debian/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list
sudo apt-get update -qq
sudo apt-get install -y -qq rocm-dev
# AMD needs this rocm_path thing exported...
export_var ROCM_PATH "/opt/rocm-4.1.0"
export_var ROCM_PATH "/opt/rocm"
export_var HIP_CXX "$ROCM_PATH/bin/hipcc"
verify_bin_exists "$HIP_CXX"
"$HIP_CXX" --version