From 94933c5234d13fe3aed930055a1edc3e7a61bace Mon Sep 17 00:00:00 2001
From: Uwe Korn <uwe.korn@quantco.com>
Date: Wed, 18 Dec 2024 18:22:59 +0000
Subject: [PATCH] Keep using our toolchain for cuda

---
 recipe/build.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/recipe/build.sh b/recipe/build.sh
index 1432152..1b11268 100644
--- 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}