From abed2b7b9289d0a2bbcc1e99c177487bf483bbc7 Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <pavol@rusnak.io>
Date: Sun, 2 Mar 2025 10:22:40 +0100
Subject: [PATCH] katago: remove cudatoolkit from deps

---
 pkgs/by-name/ka/katago/package.nix | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/pkgs/by-name/ka/katago/package.nix b/pkgs/by-name/ka/katago/package.nix
index e3ca8cd86781..f4b32cdeae8e 100644
--- a/pkgs/by-name/ka/katago/package.nix
+++ b/pkgs/by-name/ka/katago/package.nix
@@ -67,16 +67,23 @@ stdenv'.mkDerivation rec {
       boost
     ]
     ++ lib.optionals (backend == "eigen") [ eigen ]
-    ++ lib.optionals (backend == "cuda") [
-      cudaPackages.cuda_cudart
-      cudaPackages.cudnn
-      cudaPackages.cudatoolkit
-    ]
-    ++ lib.optionals (backend == "tensorrt") [
-      cudaPackages.cuda_cudart
-      cudaPackages.cudatoolkit
-      cudaPackages.tensorrt
-    ]
+    ++ lib.optionals (backend == "cuda") (
+      with cudaPackages;
+      [
+        cuda_cccl
+        cuda_cudart
+        cuda_nvcc
+        cudnn
+        libcublas
+      ]
+    )
+    ++ lib.optionals (backend == "tensorrt") (
+      with cudaPackages;
+      [
+        cuda_cudart
+        tensorrt
+      ]
+    )
     ++ lib.optionals (backend == "opencl") [
       opencl-headers
       ocl-icd
@@ -103,7 +110,7 @@ stdenv'.mkDerivation rec {
       cd cpp/
     ''
     + lib.optionalString (backend == "cuda" || backend == "tensorrt") ''
-      export CUDA_PATH="${cudaPackages.cudatoolkit}"
+      export CUDA_PATH="${cudaPackages.cuda_nvcc}"
       export EXTRA_LDFLAGS="-L/run/opengl-driver/lib"
     '';