From a38d053964bef0773d34b9cae5ba343feb994167 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Wed, 11 Feb 2026 17:43:30 -0600
Subject: [PATCH] Add nvjitlink to cuda-toolkit pip extras for cusparse
 compatibility

---
 dependencies.yaml             | 6 +++---
 python/cuml/pyproject.toml    | 2 +-
 python/libcuml/pyproject.toml | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index 1184d92f69..9b048fc664 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -376,12 +376,12 @@ dependencies:
               cuda: "12.*"
               use_cuda_wheels: "true"
             packages:
-              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse]==12.*
+              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse,nvjitlink]==12.*
           - matrix:
               cuda: "13.*"
               use_cuda_wheels: "true"
             packages:
-              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse]==13.*
+              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse,nvjitlink]==13.*
           # if use_cuda_wheels=false is provided, do not add dependencies on any CUDA wheels
           # (e.g. for DLFW and pip devcontainers)
           - matrix:
@@ -391,7 +391,7 @@ dependencies:
           # (just as a source of documentation, as this populates pyproject.toml in source control)
           - matrix:
             packages:
-              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse]>=12,<14
+              - cuda-toolkit[cublas,cufft,curand,cusolver,cusparse,nvjitlink]>=12,<14
   docs:
     common:
       - output_types: [conda, requirements]
diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
index 8a56ca57a4..15bc292953 100644
--- a/python/cuml/pyproject.toml
+++ b/python/cuml/pyproject.toml
@@ -81,7 +81,7 @@ license = "Apache-2.0"
 requires-python = ">=3.11"
 dependencies = [
     "cuda-python>=13.0.1,<14.0",
-    "cuda-toolkit[cublas,cufft,curand,cusolver,cusparse]>=12,<14",
+    "cuda-toolkit[cublas,cufft,curand,cusolver,cusparse,nvjitlink]>=12,<14",
     "cudf==26.4.*,>=0.0.0a0",
     "cupy-cuda13x>=13.6.0",
     "joblib>=0.11",
diff --git a/python/libcuml/pyproject.toml b/python/libcuml/pyproject.toml
index e3f5387a67..5253f5116a 100644
--- a/python/libcuml/pyproject.toml
+++ b/python/libcuml/pyproject.toml
@@ -25,7 +25,7 @@ classifiers = [
     "Environment :: GPU :: NVIDIA CUDA",
 ]
 dependencies = [
-    "cuda-toolkit[cublas,cufft,curand,cusolver,cusparse]>=12,<14",
+    "cuda-toolkit[cublas,cufft,curand,cusolver,cusparse,nvjitlink]>=12,<14",
     "libraft==26.4.*,>=0.0.0a0",
     "rapids-logger==0.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.