Toolchain fix stage4 (#5233)

* Update install_libtorch.sh

update right  `pkg_install_dir`

* update libri version default in build*.sh

* Update install_libtorch.sh

Fix check of libtorch.zip

Author: QuantumMisaka

toolchain/build_abacus_gnu.sh

@@ -29,8 +29,8 @@ FFTW3=$INSTALL_DIR/fftw-3.3.10
 CEREAL=$INSTALL_DIR/cereal-1.3.2/include/cereal
 LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd

toolchain/build_abacus_intel-mpich.sh

@@ -29,8 +29,8 @@ LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd
 
 cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \

toolchain/build_abacus_intel.sh

@@ -29,8 +29,8 @@ LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd
 
 # if use deepks and deepmd

toolchain/scripts/stage4/install_libtorch.sh

@@ -37,17 +37,17 @@ case "${with_libtorch}" in
   __INSTALL__)
     echo "==================== Installing libtorch ===================="
     dirname="libtorch-${libtorch_ver}"
-    filename="${dirname}.zip"
-    pkg_install_dir="${INSTALLDIR}/${filename}"
+    pkg_install_dir="${INSTALLDIR}/${dirname}"
     #pkg_install_dir="${HOME}/lib/libtorch/${libtorch_ver}"
     install_lock_file="${pkg_install_dir}/install_successful"
     archive_file="libtorch-cxx11-abi-shared-with-deps-${libtorch_ver}%2Bcpu.zip"
+    filename="${dirname}.zip"
 
     if verify_checksums "${install_lock_file}"; then
       echo "${filename} is already installed, skipping it."
     else
-        if [ -f ${archive_file} ]; then
-            echo "${archive_file} is found"
+        if [ -f ${filename} ]; then
+            echo "${filename} is found"
         else
             # download from pytorch.com and checksum
             url=https://download.pytorch.org/libtorch/cpu/${archive_file}