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

Open
aris134 wants to merge 13 commits into
base: gfx1250
Choose a base branch
from
Open

CK Tile MXFP8 Group GEMM gfx1250 #578

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 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
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
8 changes: 6 additions & 2 deletions tests/cpp/operator/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,15 +31,17 @@ list(APPEND test_cuda_sources
test_multi_unpadding.cu
test_causal_softmax.cu
test_swap_first_dims.cu
../test_common.cu)
../test_common.cu)

if(USE_CUDA)
list(APPEND test_cuda_sources
test_cast_float8blockwise.cu
test_swizzle.cu)
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 +56,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