File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 7474ROCM_VERSION_WITH_PATCH=rocm${ROCM_VERSION_MAJOR} .${ROCM_VERSION_MINOR} .${ROCM_VERSION_PATCH}
7575ROCM_INT=$(( $ROCM_VERSION_MAJOR * 10000 + $ROCM_VERSION_MINOR * 100 + $ROCM_VERSION_PATCH ))
7676
77+ PYTORCH_VERSION=$( cat $PYTORCH_ROOT /version.txt | grep -oP " [0-9]+\.[0-9]+\.[0-9]+" )
78+
7779do_lightweight_build () {
7880 echo " === Building LIGHTWEIGHT variant ==="
7981
@@ -159,7 +161,10 @@ do_heavyweight_build() {
159161 if [[ $ROCM_INT -ge 60200 ]]; then
160162 HEAVYWEIGHT_ROCM_SO_FILES+=(" librocm-core.so" )
161163 fi
162-
164+
165+ if [[ $( ver $PYTORCH_VERSION ) -ge $( ver 2.8) ]]; then
166+ HEAVYWEIGHT_ROCM_SO_FILES+=(" libhipsparselt.so" )
167+ fi
163168
164169
165170 OS_NAME=` awk -F= ' /^NAME/{print $2}' /etc/os-release`
@@ -339,7 +344,6 @@ ver() {
339344# Add triton install dependency
340345# No triton dependency till pytorch 2.3 on 3.12
341346# since torch.compile doesn't work.
342- PYTORCH_VERSION=$( cat $PYTORCH_ROOT /version.txt | grep -oP " [0-9]+\.[0-9]+\.[0-9]+" )
343347# Assuming PYTORCH_VERSION=x.y.z, if x >= 2
344348if [ ${PYTORCH_VERSION%% \. * } -ge 2 ]; then
345349 if [[ $( uname) == " Linux" ]] && [[ " $DESIRED_PYTHON " != " 3.12" || $( ver $PYTORCH_VERSION ) -ge $( ver 2.4) ]]; then
You can’t perform that action at this time.
0 commit comments