build(wheel): copy libcupti.so

GitOrigin-RevId: 8ff807012dc6ae614909fe9ae54cd4faad66e60b

scripts/whl/manylinux2014/build_wheel_common.sh

@@ -60,7 +60,9 @@ fi
 echo "Build with ${SDK_NAME}"
 
 if [ $SDK_NAME == "cu101" ];then
-    CUDA_COPY_LIB_LIST="${CUDA_LIB_DIR}/libnvrtc.so.10.1"
+    CUDA_COPY_LIB_LIST="\
+        ${CUDA_LIB_DIR}/libnvrtc.so.10.1:\
+        ${CUDA_LIB_DIR}/libcupti.so.10.1"
     EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=OFF -DMGE_WITH_CUBLAS_SHARED=OFF"
     BUILD_GCC8="ON"
     REQUIR_CUDA_VERSION="10010"
@@ -85,13 +87,8 @@ elif [ $SDK_NAME == "cu102_JetsonNano" ];then
         ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\
-        ${CUDNN_LIB_DIR}/libcudnn.so.8"
-
-    if [ ${machine} == "aarch64" ];then
-        CUDA_COPY_LIB_LIST="\
-        ${CUDA_LIB_DIR}/libcupti.so.10.2:\
-        ${CUDA_COPY_LIB_LIST}"
-    fi
+        ${CUDNN_LIB_DIR}/libcudnn.so.8:\
+        ${CUDA_LIB_DIR}/libcupti.so.10.2"
 
     EXTRA_CMAKE_FLAG="-DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_53,code=sm_53\" "
 
@@ -122,13 +119,8 @@ elif [ $SDK_NAME == "cu111" ];then
         ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\
-        ${CUDNN_LIB_DIR}/libcudnn.so.8"
-
-    if [ ${machine} == "aarch64" ];then
-        CUDA_COPY_LIB_LIST="\
-        ${CUDA_LIB_DIR}/libcupti.so.11.1:\
-        ${CUDA_COPY_LIB_LIST}"
-    fi
+        ${CUDNN_LIB_DIR}/libcudnn.so.8:\
+        ${CUDNN_LIB_DIR}/libcupti.so.11.1"
 
     if [ ${IN_CI} = "true" ] && [ ${machine} == "aarch64" ]; then
         EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_75,code=sm_75\" "
@@ -154,7 +146,8 @@ elif [ $SDK_NAME == "cu112" ];then
         ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\
         ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\
-        ${CUDNN_LIB_DIR}/libcudnn.so.8"
+        ${CUDNN_LIB_DIR}/libcudnn.so.8:\
+        ${CUDNN_LIB_DIR}/libcupti.so.11.2"
 
     EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \
         -DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \