diff --git a/dependencies.yaml b/dependencies.yaml index 605705b4ef..cc9afe3eed 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -333,12 +333,12 @@ dependencies: cuda: "12.*" use_cuda_wheels: "true" packages: - - cuda-toolkit[cublas,curand,cusolver,cusparse]==12.* + - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]==12.* - matrix: cuda: "13.*" use_cuda_wheels: "true" packages: - - cuda-toolkit[cublas,curand,cusolver,cusparse]==13.* + - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]==13.* - matrix: use_cuda_wheels: "false" packages: @@ -346,7 +346,7 @@ dependencies: # (just as a source of documentation, as this populates pyproject.toml in source control) - matrix: packages: - - cuda-toolkit[cublas,curand,cusolver,cusparse]>=12,<14 + - cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]>=12,<14 depends_on_cupy: common: - output_types: conda diff --git a/python/libcuvs/pyproject.toml b/python/libcuvs/pyproject.toml index 725e8c51f8..f43bc35dbf 100644 --- a/python/libcuvs/pyproject.toml +++ b/python/libcuvs/pyproject.toml @@ -19,7 +19,7 @@ authors = [ license = "Apache-2.0" requires-python = ">=3.11" dependencies = [ - "cuda-toolkit[cublas,curand,cusolver,cusparse]>=12,<14", + "cuda-toolkit[cublas,curand,cusolver,cusparse,nvjitlink]>=12,<14", "libraft==26.4.*,>=0.0.0a0", "librmm==26.4.*,>=0.0.0a0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.