[CK_TILE] MX Flatmm Use Byte Pointer Arithmetic for A Tensor (#3446)

* A as bytes

* Reformat with static_for_product
This commit is contained in:
Yi DING
2025-12-19 10:28:13 +08:00
committed by GitHub
parent c0ee71d735
commit 2220cbaba7
4 changed files with 309 additions and 314 deletions

View File

@@ -20,6 +20,7 @@ if(has_supported_gpu)
if(CK_USE_OCP_FP8)
list(APPEND EXAMPLE_FLATMM_COMPILE_OPTIONS -DCK_TILE_USE_OCP_FP8)
endif()
list(APPEND EXAMPLE_FLATMM_COMPILE_OPTIONS "SHELL: -mllvm -greedy-reverse-local-assignment=1")
add_executable(tile_example_flatmm_basic flatmm_basic.cpp)
target_compile_options(tile_example_flatmm_basic PRIVATE ${EXAMPLE_FLATMM_COMPILE_OPTIONS})