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",
@@ -11,8 +11,8 @@ from cutlass.cutlass_dsl import dsl_user_op
NVVM_CTA_GROUP_MAP = [
None,
nvvm.Tcgen05GroupKind.CTA_1,
nvvm.Tcgen05GroupKind.CTA_2,
nvvm.CTAGroupKind.CTA_1,
nvvm.CTAGroupKind.CTA_2,
]
LDST_MAP = {
"32x32b": (nvvm.Tcgen05LdStShape.SHAPE_32X32B, 1),
@@ -134,7 +134,7 @@ def commit(mbar, cta_mask=None, cta_group: int = 1, *, loc=None, ip=None):
mbar_llvm = mbar.to_llvm_ptr(loc=loc, ip=ip)
group = NVVM_CTA_GROUP_MAP[cta_group]
if cutlass.const_expr(cta_mask is not None):
nvvm.tcgen05_commit_arrive(
nvvm.tcgen05_commit(
mbar_llvm,
multicast_mask=cta_mask.ir_value(loc=loc, ip=ip),
group=group,
@@ -142,7 +142,7 @@ def commit(mbar, cta_mask=None, cta_group: int = 1, *, loc=None, ip=None):
ip=ip,
)
else:
nvvm.tcgen05_commit_arrive(mbar_llvm, group=group, loc=loc, ip=ip)
nvvm.tcgen05_commit(mbar_llvm, group=group, loc=loc, ip=ip)
@dsl_user_op
@@ -34,7 +34,7 @@ if TYPE_CHECKING:
logger = init_logger(__name__)
SGL_TEST_FILES_CI_DATA_REVISION = "d51ca9623e0bb27087da243a44c942fdda5aafe5"
SGL_TEST_FILES_CI_DATA_REVISION = "320949ecc2587474a2f535229ffc8f47ed16ee51"
if current_platform.is_npu():
SGL_TEST_FILES_CI_DATA_REVISION = "6b62f4b6825c76a25fd2ba28248df68f2b400e65"
+1 -1
View File
@@ -1820,7 +1820,7 @@ def suppress_noisy_warnings():
cutlass_dsl_noisy = {
(
DeprecationWarning,
"Use explicit `struct.scalar.ptr` for pointer instead.",
"Using `struct.scalar` as pointer is deprecated.",
),
(
UserWarning,