Skip to content

Commit 5f70bd2

Browse files
committed
[None][fix] Use legal MXF8 scale-vector encoding
Signed-off-by: Mingyang Hao <200044211+mingyangHao@users.noreply.github.com>
1 parent 8348419 commit 5f70bd2

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

tensorrt_llm/_torch/cute_dsl_kernels/blackwell/dynamic_mxf8_mainloop.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,7 @@
1111
from cutlass.cute.arch.nvvm_wrappers import nvvm
1212
from cutlass.cutlass_dsl import Boolean, Int32, Uint32, dsl_user_op
1313

14-
_MXF8_SCALE_VEC_SIZE = getattr(
15-
_nvvm_raw.Tcgen05MMAScaleVecSize,
16-
"BLOCK32",
17-
_nvvm_raw.Tcgen05MMAScaleVecSize.X1,
18-
)
14+
_MXF8_SCALE_VEC_SIZE = _nvvm_raw.Tcgen05MMAScaleVecSize.X1
1915

2016

2117
def _align16(x):

0 commit comments

Comments
 (0)