Bump CuTe DSL to 4.6.0 (#31714)
This commit is contained in:
@@ -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"
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user