minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Bump CuTeDSL to 4.6.2 (#34372)

Browse Source
This commit is contained in:
Mohammad Miadh Angkad
2026-08-12 00:35:35 +08:00
committed by GitHub
parent 8267d76c2c
commit a3bd7d9401
3 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/pyproject.toml
+2 -2
View File
@@ -46,7 +46,7 @@ dependencies = [
"ninja", "ninja",
"numba==0.65.1", "numba==0.65.1",
"numpy", "numpy",
"nvidia-cutlass-dsl[cu13]==4.6.0", "nvidia-cutlass-dsl[cu13]==4.6.2",
"nvidia-mathdx==25.6.0", "nvidia-mathdx==25.6.0",
"nvidia-ml-py", "nvidia-ml-py",
"openai==2.6.1", "openai==2.6.1",
@@ -63,7 +63,7 @@ dependencies = [
"pydantic", "pydantic",
"python-multipart", "python-multipart",
"pyzmq>=25.1.2", "pyzmq>=25.1.2",
"quack-kernels>=0.6.1", "quack-kernels==0.6.4",
"requests", "requests",
"scipy", "scipy",
"sentencepiece", "sentencepiece",
python/sglang/kernels/ops/attention/flash_attn/cute/interface.py
-2
View File
@@ -1,6 +1,4 @@
# Copyright (c) 2025, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao. # Copyright (c) 2025, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao.
# [2025-07-04] Version in Cute-DSL, for Hopper and Blackwell. You'll need install nvidia-cutlass-dsl==4.2.0.
import math import math
import os import os
from dataclasses import dataclass from dataclasses import dataclass
python/sglang/kernels/ops/attention/flash_attn/cute/pyproject.toml
+3 -3
View File
@@ -22,17 +22,17 @@ classifiers = [
] ]
dependencies = [ dependencies = [
"nvidia-cutlass-dsl>=4.5.2", "nvidia-cutlass-dsl==4.6.2",
"torch", "torch",
"einops", "einops",
"typing_extensions", "typing_extensions",
"apache-tvm-ffi>=0.1.5,<0.2", "apache-tvm-ffi>=0.1.5,<0.2",
"torch-c-dlpack-ext", "torch-c-dlpack-ext",
"quack-kernels>=0.5.0", "quack-kernels==0.6.4",
] ]
[project.optional-dependencies] [project.optional-dependencies]
cu13 = ["nvidia-cutlass-dsl[cu13]>=4.5.2"] cu13 = ["nvidia-cutlass-dsl[cu13]==4.6.2"]
dev = [ dev = [
"pytest", "pytest",
"pytest-xdist", "pytest-xdist",