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

* A as bytes

* Reformat with static_for_product

[ROCm/composable_kernel commit: 2220cbaba7]
This commit is contained in:
Yi DING
2025-12-19 10:28:13 +08:00
committed by GitHub
parent f1b3ca26b3
commit 1fc7bdd402
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})