Keep using our toolchain for cuda

conda-forge · Dec 18, 2024 · 94933c5 · 94933c5
1 parent 73b2fe3
commit 94933c5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/recipe/build.sh b/recipe/build.sh
@@ -118,7 +118,10 @@ if [[ "${target_platform}" == linux-* ]]; then
     sed -i '/Qunused-arguments/d' .bazelrc
     # Don't override our toolchain for CUDA
     sed -i '/TF_NVCC_CLANG/{N;d}' .bazelrc
+    # Keep using our toolchain
+    sed -i '/--crosstool_top=@local_config_cuda/d' .bazelrc
 fi
+
 ${PYTHON} build/build.py build \
     --target_cpu_features default \
     ${EXTRA}