Skip to content

CK Tile MXFP8 Group GEMM gfx1250 #578

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
aris134 merged 20 commits into from
May 21, 2026
Merged

CK Tile MXFP8 Group GEMM gfx1250 #578

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
1f707d7
initial commit for CK Tile MXFP8 integration for gfx1250
aris134 Apr 25, 2026
e102f00
ck mxfp8 gfx1250 integration builds successfully
aris134 Apr 25, 2026
52a2887
add entrypoint to ck mx group gemm in caller
aris134 Apr 25, 2026
8022777
temporary hacky change to test_numerics for bringup testing
aris134 Apr 25, 2026
bc6253d
add warning print to confirm we are in fallback
aris134 Apr 25, 2026
d26f52e
MXFP8 grouped fwd/bwd now reaches CK path and runs without fallback/c…
aris134 May 2, 2026
e295e74
add cpp test for ck tile group mxfp8 gemm forward
aris134 May 2, 2026
1784045
Fix MXFP8 grouped GEMM scale handling for NN/TN/NT
aris134 May 3, 2026
fe99bf3
update ck mxfp8 group gemm gtest to exercise mixed dtypes
aris134 May 3, 2026
e7159c4
include renamed test file
aris134 May 3, 2026
972cea3
clean up code
aris134 May 3, 2026
c0fabff
Update cublaslt_gemm.cu
aris134 May 6, 2026
3db2e5a
address pr comments
aris134 May 11, 2026
910d30f
fix ck group mxfp8 dispatch
aris134 May 17, 2026
1b66d29
update CMakeLists.txt
aris134 May 17, 2026
23b505f
Add direct ROCm libraries dependency for CK grouped GEMM
aris134 May 19, 2026
746afea
Remove redundant MXFP8 env override from grouped linear test
aris134 May 19, 2026
175855d
factor out common definitions from mxfp8 ck ggemm
aris134 May 20, 2026
f00fb7f
add pr comments
aris134 May 21, 2026
74744db
Merge remote-tracking branch 'origin/gfx1250' into amartin/ck-mxfp8-g…
aris134 May 21, 2026
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
5 changes: 4 additions & 1 deletion tests/cpp/operator/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,8 @@ if(USE_CUDA)
else()
list(APPEND test_cuda_sources
test_cublaslt_gemm.cu
test_cast_mxfp4_transpose.cu)
test_cast_mxfp4_transpose.cu
test_ck_grouped_mxfp8.cu)
endif()

if(USE_CUDA)
Expand All @@ -54,12 +55,14 @@ endif()

# Find required packages
find_package(OpenMP REQUIRED)

if(USE_CUDA)
list(APPEND test_operator_LINKER_LIBS CUDA::cudart GTest::gtest_main ${TE_LIB} CUDA::nvrtc CUDNN::cudnn)
target_link_libraries(test_operator PUBLIC ${test_operator_LINKER_LIBS} OpenMP::OpenMP_CXX)
else()
target_link_libraries(test_operator PUBLIC hip::host hip::device GTest::gtest_main ${TE_LIB} OpenMP::OpenMP_CXX hiprand)
endif()

target_compile_options(test_operator PRIVATE -O2 -fopenmp)

include(GoogleTest)
Expand Down
Loading
Loading