diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index c9d0341..232347b 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -9,7 +9,7 @@ repos: hooks: - id: clang-format - repo: https://github.com/rapidsai/dependency-file-generator - rev: v1.13.11 + rev: v1.16.0 hooks: - id: rapids-dependency-file-generator args: ["--clean"] @@ -19,7 +19,7 @@ repos: - id: ruff - id: ruff-format - repo: https://github.com/rapidsai/pre-commit-hooks - rev: v0.1.0 + rev: v0.4.0 hooks: - id: verify-copyright args: [--fix, --main-branch=main] diff --git a/ci/build_conda.sh b/ci/build_conda.sh index 946a95d..5c75d85 100755 --- a/ci/build_conda.sh +++ b/ci/build_conda.sh @@ -24,9 +24,13 @@ cuda_compiler_version: - "${CUDA_VERSION}" EOF +sccache --zero-stats + rapids-conda-retry build \ conda/recipes/pynvjitlink \ -m cuda_compiler_version.yaml \ ; +sccache --show-adv-stats + rapids-upload-conda-to-s3 python diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index 31d0017..4f8ec80 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -3,17 +3,23 @@ set -euo pipefail +source rapids-configure-sccache + rapids-logger "Install CUDA Toolkit" source "$(dirname "$0")/install_latest_cuda_toolkit.sh" RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})" +sccache --zero-stats + rapids-logger "Build wheel" mkdir -p ./dist -python -m pip wheel . --wheel-dir=./dist -vvv --disable-pip-version-check --no-deps +python -m pip wheel . --wheel-dir=./dist -v --disable-pip-version-check --no-deps + +sccache --show-adv-stats # Exclude libcuda.so.1 because we only install a driver stub python -m auditwheel repair --exclude libcuda.so.1 -w ./final_dist ./dist/* rapids-logger "Upload Wheel" -RAPIDS_PY_WHEEL_NAME="pynvjitlink_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 ./final_dist +RAPIDS_PY_WHEEL_NAME="pynvjitlink_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 python ./final_dist diff --git a/conda/recipes/pynvjitlink/meta.yaml b/conda/recipes/pynvjitlink/meta.yaml index e0a3502..d5cb6b1 100644 --- a/conda/recipes/pynvjitlink/meta.yaml +++ b/conda/recipes/pynvjitlink/meta.yaml @@ -18,6 +18,22 @@ source: build: script: - {{ PYTHON }} -m pip install . -vv + script_env: + - AWS_ACCESS_KEY_ID + - AWS_SECRET_ACCESS_KEY + - AWS_SESSION_TOKEN + - CMAKE_C_COMPILER_LAUNCHER + - CMAKE_CUDA_COMPILER_LAUNCHER + - CMAKE_CXX_COMPILER_LAUNCHER + - CMAKE_GENERATOR + - PARALLEL_LEVEL + - SCCACHE_BUCKET + - SCCACHE_IDLE_TIMEOUT + - SCCACHE_REGION + - SCCACHE_S3_KEY_PREFIX=pynvjitlink-aarch64 # [aarch64] + - SCCACHE_S3_KEY_PREFIX=pynvjitlink-linux64 # [linux64] + - SCCACHE_S3_USE_SSL + - SCCACHE_S3_NO_CREDENTIALS ignore_run_exports_from: - {{ compiler('cuda') }} - libnvjitlink-dev