From 59a2ddb78bbf7486ac6274a5b4c03be27295d177 Mon Sep 17 00:00:00 2001 From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com> Date: Mon, 18 May 2026 14:05:57 +0800 Subject: [PATCH] Use cu13 extra for nvidia cutlass dsl Switch nvidia-cutlass-dsl to the cu13 extra and update the pin to 4.5.1 for the CUDA 13 default environment. --- python/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index d5c2f3d6b703..d52c17000615 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -37,7 +37,7 @@ dependencies = [ "ninja", "easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports "numpy", - "nvidia-cutlass-dsl[cu13]==4.5.0", + "nvidia-cutlass-dsl[cu13]==4.5.1", "nvidia-ml-py", "openai-harmony==0.0.4", "openai==2.6.1",