Bump CuTe DSL to 4.6.0 (#31714)

This commit is contained in:
Mohammad Miadh Angkad
2026-07-20 02:11:59 -07:00
committed by GitHub
parent 97e0647bdc
commit 3d82dacd58
5 changed files with 9 additions and 38 deletions
+3 -3
View File
@@ -30,7 +30,7 @@ dependencies = [
"easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
"einops",
"fastapi",
"flash-attn-4==4.0.0b15",
"flash-attn-4>=4.0.0b18",
"flashinfer_python[cu13]==0.6.14", # keep it aligned with jit-cache version in Dockerfile
"gguf",
"helion==0.2.6",
@@ -45,7 +45,7 @@ dependencies = [
"ninja",
"numba==0.65.1",
"numpy",
"nvidia-cutlass-dsl[cu13]==4.5.2",
"nvidia-cutlass-dsl[cu13]==4.6.0",
"nvidia-mathdx==25.6.0",
"nvidia-ml-py",
"openai==2.6.1",
@@ -62,7 +62,7 @@ dependencies = [
"pydantic",
"python-multipart",
"pyzmq>=25.1.2",
"quack-kernels>=0.4.1",
"quack-kernels>=0.6.1",
"requests",
"scipy",
"sentencepiece",