try using sccache

rapidsai · Oct 29, 2024 · b99d8ca · b99d8ca
1 parent 2bb05d8
commit b99d8ca
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -3,6 +3,8 @@
 
 set -euo pipefail
 
+source rapids-configure-sccache
+
 rapids-logger "Install CUDA Toolkit"
 source "$(dirname "$0")/install_latest_cuda_toolkit.sh"
 

diff --git 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