update ci

Author: OutisLi

.github/workflows/build_cc.yml

@@ -38,7 +38,7 @@ jobs:
           wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \
           && sudo dpkg -i cuda-keyring_1.0-1_all.deb \
           && sudo apt-get update \
-          && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2
+          && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2 libcublas-dev-12-2 libcusparse-dev-12-2
         if: matrix.variant == 'cuda120'
         env:
           DEBIAN_FRONTEND: noninteractive

.github/workflows/codeql.yml

@@ -42,7 +42,7 @@ jobs:
           wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb \
            && sudo dpkg -i cuda-keyring_1.0-1_all.deb \
            && sudo apt-get update \
-           && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2
+           && sudo apt-get -y install cuda-cudart-dev-12-2 cuda-nvcc-12-2 libcublas-dev-12-2 libcusparse-dev-12-2
           python -m pip install uv
           uv pip install --system --group pin_tensorflow_cpu --group pin_pytorch_cpu --torch-backend cpu
         env:

pyproject.toml

@@ -305,7 +305,7 @@ environment-pass = [
 ]
 before-all = [
     """if [ ! -z "${DP_PKG_NAME}" ]; then sed -i "s/name = \\"deepmd-kit\\"/name = \\"${DP_PKG_NAME}\\"/g" pyproject.toml; fi""",
-    """{ if [ -n "${CUDA_VERSION}" ] && [ "$(uname -m)" = "x86_64" ] ; then yum config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo && yum install -y cuda-nvcc-${CUDA_VERSION/./-} cuda-cudart-devel-${CUDA_VERSION/./-}; fi }""",
+    """{ if [ -n "${CUDA_VERSION}" ] && [ "$(uname -m)" = "x86_64" ] ; then yum config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo && yum install -y cuda-nvcc-${CUDA_VERSION/./-} cuda-cudart-devel-${CUDA_VERSION/./-} libcublas-devel-${CUDA_VERSION/./-} libcusparse-devel-${CUDA_VERSION/./-}; fi }""",
 ]
 before-build = [
 ]

source/install/docker_package_c.sh

@@ -6,11 +6,17 @@ PYTHON_BIN=${PYTHON_BIN:-/opt/python/cp311-cp311/bin/python}
 PYTORCH_DEPENDENCY_GROUP=${PYTORCH_DEPENDENCY_GROUP:-pin_pytorch_cpu}
 PYTORCH_TORCH_BACKEND=${PYTORCH_TORCH_BACKEND:-cpu}
 TENSORFLOW_DEPENDENCY_GROUP=${TENSORFLOW_DEPENDENCY_GROUP:-pin_tensorflow_cpu}
+# CUDA version tag (e.g. "12-9") parsed from the manylinux CUDA image name. It
+# selects the cuBLAS/cuSPARSE development packages installed inside the
+# container: the fused PyTorch CUDA operators pull in ATen CUDA headers that
+# require cublas and cusparse headers absent from the base image.
+CUDA_VERSION_DASH=$(printf '%s' "${MANYLINUX_CUDA_IMAGE}" | sed -n 's/.*cuda\([0-9][0-9]*\)_\([0-9][0-9]*\).*/\1-\2/p')
 
 docker run --rm -v "${SCRIPT_PATH}/../..":/root/deepmd-kit -w /root/deepmd-kit \
 	-e CIBUILDWHEEL="${CIBUILDWHEEL:-1}" \
 	-e ENABLE_PYTORCH="${ENABLE_PYTORCH:-FALSE}" \
 	-e MANYLINUX_CUDA_IMAGE="${MANYLINUX_CUDA_IMAGE}" \
+	-e CUDA_VERSION_DASH="${CUDA_VERSION_DASH}" \
 	-e PACKAGE_C_RUNTIME_DEPENDENCY_PRE_EXCLUDE_REGEXES="${PACKAGE_C_RUNTIME_DEPENDENCY_PRE_EXCLUDE_REGEXES:-}" \
 	-e PACKAGE_C_RUNTIME_DEPENDENCY_POST_EXCLUDE_REGEXES="${PACKAGE_C_RUNTIME_DEPENDENCY_POST_EXCLUDE_REGEXES:-}" \
 	-e PYTORCH_DEPENDENCY_GROUP="${PYTORCH_DEPENDENCY_GROUP}" \
@@ -20,6 +26,10 @@ docker run --rm -v "${SCRIPT_PATH}/../..":/root/deepmd-kit -w /root/deepmd-kit \
 	-e TENSORFLOW_DEPENDENCY_GROUP="${TENSORFLOW_DEPENDENCY_GROUP}" \
 	"${MANYLINUX_CUDA_IMAGE}" \
 	/bin/bash -lc 'set -euo pipefail
+            if [ -n "${CUDA_VERSION_DASH:-}" ]; then
+              yum config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
+              yum install -y "libcublas-devel-${CUDA_VERSION_DASH}" "libcusparse-devel-${CUDA_VERSION_DASH}"
+            fi
             export PATH="$(dirname "${PYTHON_BIN}"):${PATH}"
             "${PYTHON_BIN}" -m pip install uv
             UV_INSTALL_ARGS=(