From c2f6e1f04b4133cfd445abafcf68ba2d8fbbc58c Mon Sep 17 00:00:00 2001 From: Kaustubh Tangsali <71059996+ktangsali@users.noreply.github.com> Date: Tue, 10 Feb 2026 13:37:04 -0800 Subject: [PATCH 1/2] Upgrade GPU dependencies in pyproject.toml to CUDA 13 Updated GPU dependencies to newer versions. --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index b0ecaaf..263cbbb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,8 +33,8 @@ dynamic = ["version"] [project.optional-dependencies] # dependencies for GPU-accelerated functionality gpu = [ - "cupy-cuda12x>=13.3.0", - "cuml-cu12>=25.2.0", + "cupy-cuda13x>=13.6.0", + "cuml-cu13>=25.10.0", "warp-lang>=1.7.1", ] all = [ From 944b53f0797764579c608fb3eaee4c23c3b45b20 Mon Sep 17 00:00:00 2001 From: Kaustubh Tangsali <71059996+ktangsali@users.noreply.github.com> Date: Tue, 10 Feb 2026 13:41:37 -0800 Subject: [PATCH 2/2] Update requirements for cupy and cuml packages --- workflows/domino_design_sensitivities/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/workflows/domino_design_sensitivities/requirements.txt b/workflows/domino_design_sensitivities/requirements.txt index 02e4d67..611c458 100644 --- a/workflows/domino_design_sensitivities/requirements.txt +++ b/workflows/domino_design_sensitivities/requirements.txt @@ -1,6 +1,6 @@ git+https://github.com/NVIDIA/apex.git hydra-core -cupy-cuda12x +cupy-cuda13x numba requests -"cuml-cu12>=25.2.0" \ No newline at end of file +"cuml-cu13>=25.10.0"