Skip to content

Commit

Permalink
Allow ROCm minor releases to use the same MIOpen branch as the major …
Browse files Browse the repository at this point in the history
…release (pytorch#14)
  • Loading branch information
jithunnair-amd authored Oct 24, 2022
1 parent 55db5ba commit 594e241
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions common/install_miopen.sh
Original file line number Diff line number Diff line change
Expand Up @@ -59,26 +59,26 @@ MIOPEN_CMAKE_COMMON_FLAGS="
-DMIOPEN_BUILD_DRIVER=OFF
"
# Pull MIOpen repo and set DMIOPEN_EMBED_DB based on ROCm version
if [[ $ROCM_INT -eq 50300 ]]; then
if [[ $ROCM_INT -ge 50300 ] && [ $ROCM_INT -lt 50400 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx906_60;gfx90878;gfx90a6e;gfx1030_36 -DMIOPEN_USE_MLIR=Off"
MIOPEN_BRANCH="release/rocm-rel-5.3-staging"
elif [[ $ROCM_INT -eq 50200 ]]; then
elif [[ $ROCM_INT -ge 50200 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx906_60;gfx90878;gfx90a6e;gfx1030_36 -DMIOPEN_USE_MLIR=Off"
MIOPEN_BRANCH="release/rocm-rel-5.2-staging"
elif [[ $ROCM_INT -eq 50100 ]]; then
elif [[ $ROCM_INT -ge 50100 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx906_60;gfx90878;gfx90a6e;gfx1030_36"
MIOPEN_BRANCH="release/rocm-rel-5.1-staging"
elif [[ $ROCM_INT -eq 50000 ]]; then
elif [[ $ROCM_INT -ge 50000 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx906_60;gfx90878;gfx90a6e;gfx1030_36"
MIOPEN_BRANCH="release/rocm-rel-5.0-staging"
elif [[ $ROCM_INT -eq 40500 ]]; then
elif [[ $ROCM_INT -ge 40500 ]]; then
MIOPEN_CMAKE_COMMON_FLAGS="${MIOPEN_CMAKE_COMMON_FLAGS} -DMIOPEN_USE_HIP_KERNELS=Off -DMIOPEN_DEFAULT_FIND_MODE=Normal"
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx906_60;gfx90878;gfx90a6e;gfx1030_36"
MIOPEN_BRANCH="release/rocm-rel-4.5-staging"
elif [[ $ROCM_INT -eq 40300 ]]; then
elif [[ $ROCM_INT -ge 40300 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx900_56;gfx900_64;gfx906_60;gfx906_64;gfx90878;gfx1030_36"
MIOPEN_BRANCH="release/rocm-rel-4.3-staging"
elif [[ $ROCM_INT -eq 40200 ]]; then
elif [[ $ROCM_INT -ge 40200 ]]; then
MIOPEN_CMAKE_DB_FLAGS="-DMIOPEN_EMBED_DB=gfx803_36;gfx803_64;gfx900_56;gfx900_64;gfx906_60;gfx906_64;gfx90878"
MIOPEN_BRANCH="rocm-4.2.x-staging"
else
Expand Down

0 comments on commit 594e241

Please sign in to comment.