Remove nvprune as libs are from pypi

tinglvv · tinglvv · commit f1227b3ccb52 · 2025-01-24T12:31:35.000-08:00
diff --git a/.ci/docker/common/install_cuda.sh b/.ci/docker/common/install_cuda.sh
@@ -347,40 +347,6 @@ function install_128 {
   ldconfig
 }
 
-function prune_128 {
-  echo "Pruning CUDA 12.8"
-  #####################################################################################
-  # CUDA 12.8 prune static libs
-  #####################################################################################
-  export NVPRUNE="/usr/local/cuda-12.8/bin/nvprune"
-  export CUDA_LIB_DIR="/usr/local/cuda-12.8/lib64"
-
-  export GENCODE="-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120"
-  export GENCODE_CUDNN="-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120"
-
-  if [[ -n "$OVERRIDE_GENCODE" ]]; then
-      export GENCODE=$OVERRIDE_GENCODE
-  fi
-  if [[ -n "$OVERRIDE_GENCODE_CUDNN" ]]; then
-      export GENCODE_CUDNN=$OVERRIDE_GENCODE_CUDNN
-  fi
-
-  # all CUDA static libs except CuDNN and CuBLAS
-  ls $CUDA_LIB_DIR/ | grep "\.a$" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis"  \
-      | xargs -I {} bash -c \
-                "echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/:x{} -o $CUDA_LIB_DIR/{}"
-
-  # prune CuDNN and CuBLAS
-  $NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
-  $NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
-
-  #####################################################################################
-  # CUDA 12.8 prune visual tools
-  #####################################################################################
-  export CUDA_BASE="/usr/local/cuda-12.8/"
-  rm -rf $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2025.1.0 $CUDA_BASE/nsight-systems-2024.6.2/
-}
-
 # idiomatic parameter and option handling in sh
 while test $# -gt 0
 do
@@ -391,7 +357,7 @@ do
         ;;
     12.6) install_126; prune_126
         ;;
-    12.8) install_128; prune_128
+    12.8) install_128;
         ;;
     *) echo "bad argument $1"; exit 1
         ;;
diff --git a/.ci/docker/common/install_cuda_aarch64.sh b/.ci/docker/common/install_cuda_aarch64.sh
@@ -194,40 +194,6 @@ function install_128 {
   ldconfig
 }
 
-function prune_128 {
-  echo "Pruning CUDA 12.8"
-  #####################################################################################
-  # CUDA 12.8 prune static libs
-  #####################################################################################
-  export NVPRUNE="/usr/local/cuda-12.8/bin/nvprune"
-  export CUDA_LIB_DIR="/usr/local/cuda-12.8/lib64"
-
-  export GENCODE="-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90"
-  export GENCODE_CUDNN="-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90"
-
-  if [[ -n "$OVERRIDE_GENCODE" ]]; then
-      export GENCODE=$OVERRIDE_GENCODE
-  fi
-  if [[ -n "$OVERRIDE_GENCODE_CUDNN" ]]; then
-      export GENCODE_CUDNN=$OVERRIDE_GENCODE_CUDNN
-  fi
-
-  # all CUDA static libs except CuDNN and CuBLAS
-  ls $CUDA_LIB_DIR/ | grep "\.a$" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis"  \
-      | xargs -I {} bash -c \
-                "echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/{} -o $CUDA_LIB_DIR/{}"
-
-  # prune CuDNN and CuBLAS
-  $NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
-  $NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
-
-  #####################################################################################
-  # CUDA 12.8 prune visual tools
-  #####################################################################################
-  export CUDA_BASE="/usr/local/cuda-12.8/"
-  rm -rf $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2025.1.0 $CUDA_BASE/nsight-systems-2024.6.2/
-}
-
 # idiomatic parameter and option handling in sh
 while test $# -gt 0
 do
@@ -236,7 +202,7 @@ do
         ;;
     12.6) install_126; prune_126
         ;;
-    12.8) install_128; prune_128
+    12.8) install_128;
         ;;
     *) echo "bad argument $1"; exit 1
         ;;
