fix(cublas): add cuda-crt package for crt/host_defines.h
cublasLt.h -> cublas_api.h -> driver_types.h -> crt/host_defines.h which lives in the cuda-crt-13-0 package, not cudart-dev. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -28,6 +28,7 @@ PACKAGES_GZ="${DOWNLOAD_CACHE_DIR}/Packages.gz"
|
|||||||
echo "=== cuBLAS ${CUBLAS_VERSION} / cudart ${CUDA_USERSPACE_VERSION} / CUDA ${CUDA_SERIES} ==="
|
echo "=== cuBLAS ${CUBLAS_VERSION} / cudart ${CUDA_USERSPACE_VERSION} / CUDA ${CUDA_SERIES} ==="
|
||||||
|
|
||||||
if [ -f "${CACHE_DIR}/include/cublasLt.h" ] && [ -f "${CACHE_DIR}/include/cuda_runtime_api.h" ] \
|
if [ -f "${CACHE_DIR}/include/cublasLt.h" ] && [ -f "${CACHE_DIR}/include/cuda_runtime_api.h" ] \
|
||||||
|
&& [ -f "${CACHE_DIR}/include/crt/host_defines.h" ] \
|
||||||
&& [ "$(find "${CACHE_DIR}/lib" \( -name 'libcublas.so*' -o -name 'libcublasLt.so*' -o -name 'libcudart.so*' \) 2>/dev/null | wc -l)" -gt 0 ]; then
|
&& [ "$(find "${CACHE_DIR}/lib" \( -name 'libcublas.so*' -o -name 'libcublasLt.so*' -o -name 'libcudart.so*' \) 2>/dev/null | wc -l)" -gt 0 ]; then
|
||||||
echo "=== cuBLAS cached, skipping download ==="
|
echo "=== cuBLAS cached, skipping download ==="
|
||||||
echo "cache: $CACHE_DIR"
|
echo "cache: $CACHE_DIR"
|
||||||
@@ -155,14 +156,17 @@ CUBLAS_RT_DEB=$(download_verified_pkg "libcublas-${CUDA_SERIES_DASH}" "${CUBLAS_
|
|||||||
CUBLAS_DEV_DEB=$(download_verified_pkg "libcublas-dev-${CUDA_SERIES_DASH}" "${CUBLAS_VERSION}")
|
CUBLAS_DEV_DEB=$(download_verified_pkg "libcublas-dev-${CUDA_SERIES_DASH}" "${CUBLAS_VERSION}")
|
||||||
CUDART_RT_DEB=$(download_verified_pkg "cuda-cudart-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}")
|
CUDART_RT_DEB=$(download_verified_pkg "cuda-cudart-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}")
|
||||||
CUDART_DEV_DEB=$(download_verified_pkg "cuda-cudart-dev-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}")
|
CUDART_DEV_DEB=$(download_verified_pkg "cuda-cudart-dev-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}")
|
||||||
|
CUDA_CRT_DEB=$(download_verified_pkg "cuda-crt-${CUDA_SERIES_DASH}" "${CUDA_USERSPACE_VERSION}")
|
||||||
|
|
||||||
extract_deb "$CUBLAS_RT_DEB" "${TMP_DIR}/cublas-rt"
|
extract_deb "$CUBLAS_RT_DEB" "${TMP_DIR}/cublas-rt"
|
||||||
extract_deb "$CUBLAS_DEV_DEB" "${TMP_DIR}/cublas-dev"
|
extract_deb "$CUBLAS_DEV_DEB" "${TMP_DIR}/cublas-dev"
|
||||||
extract_deb "$CUDART_RT_DEB" "${TMP_DIR}/cudart-rt"
|
extract_deb "$CUDART_RT_DEB" "${TMP_DIR}/cudart-rt"
|
||||||
extract_deb "$CUDART_DEV_DEB" "${TMP_DIR}/cudart-dev"
|
extract_deb "$CUDART_DEV_DEB" "${TMP_DIR}/cudart-dev"
|
||||||
|
extract_deb "$CUDA_CRT_DEB" "${TMP_DIR}/cuda-crt"
|
||||||
|
|
||||||
copy_headers "${TMP_DIR}/cublas-dev"
|
copy_headers "${TMP_DIR}/cublas-dev"
|
||||||
copy_headers "${TMP_DIR}/cudart-dev"
|
copy_headers "${TMP_DIR}/cudart-dev"
|
||||||
|
copy_headers "${TMP_DIR}/cuda-crt"
|
||||||
copy_libs "${TMP_DIR}/cublas-rt"
|
copy_libs "${TMP_DIR}/cublas-rt"
|
||||||
copy_libs "${TMP_DIR}/cudart-rt"
|
copy_libs "${TMP_DIR}/cudart-rt"
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user