From ca6b5eddaf25a56571a950b2b214883aebbde42a Mon Sep 17 00:00:00 2001
From: Break Yang <breakds@gmail.com>
Date: Tue, 20 May 2025 15:49:47 -0700
Subject: [PATCH] Replace `cudatoolkit` with only the necessary parts

---
 pkgs/development/python-modules/flashinfer/default.nix | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pkgs/development/python-modules/flashinfer/default.nix b/pkgs/development/python-modules/flashinfer/default.nix
index 7f0cd14224a9..d9f95a650a27 100644
--- a/pkgs/development/python-modules/flashinfer/default.nix
+++ b/pkgs/development/python-modules/flashinfer/default.nix
@@ -46,6 +46,7 @@ buildPythonPackage {
     cmake
     ninja
     (lib.getBin cudaPackages.cuda_nvcc)
+    (lib.getLib cudaPackages.cuda_cudart)
   ];
   dontUseCmakeConfigure = true;