Skip to content

Upgrade CUTLASS 4.0 #6336

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
zhyncs merged 4 commits into from
May 16, 2025
Merged

Upgrade CUTLASS 4.0 #6336

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
74e5a98
chore: upgrade cutlass 4.0
zhyncs May 13, 2025
378e5ee
Merge branch 'main' into zhyncs/upd
zhyncs May 15, 2025
0353ec8
modify the template for upgrading to CUTLASS 4.0
elfiegg May 15, 2025
70898e0
Merge branch 'main' into pr-6272
zhyncs May 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion sgl-kernel/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ include(FetchContent)
FetchContent_Declare(
repo-cutlass
GIT_REPOSITORY https://github.com/NVIDIA/cutlass
GIT_TAG ad7b2f5e84fcfa124cb02b91d5bd26d238c0459e
GIT_TAG f115c3f85467d5d9619119d1dbeb9c03c3d73864
GIT_SHALLOW OFF
)
FetchContent_Populate(repo-cutlass)
Expand Down
2 changes: 1 addition & 1 deletion sgl-kernel/csrc/gemm/fp8_blockwise_gemm_kernel.cu
100644 → 100755
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ void launch_sm90_fp8_blockwise_scaled_mm(
LayoutSFB layout_sfb = ScaleConfig::tile_atom_to_shape_SFB(make_shape(m, n, k, 1));

typename GemmKernel::MainloopArguments mainloop_args{
a_ptr, stride_a, b_ptr, stride_b, 4, a_s_ptr, layout_sfa, b_s_ptr, layout_sfb};
a_ptr, stride_a, b_ptr, stride_b, a_s_ptr, layout_sfa, b_s_ptr, layout_sfb};
typename GemmKernel::EpilogueArguments epilogue_args{{}, nullptr, stride_d, o_ptr, stride_d};

typename Gemm::Arguments args = {
Expand Down
Loading