mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-24 14:54:47 +00:00
b89a44ea33
* Batched GEMM Multiple D based on Universal GEMM
Co-authored-by: Jing Zhang <jizhan@fb.com>
* CI fixes
Co-authored-by: Jing Zhang <jizhan@fb.com>
---------
Co-authored-by: Jing Zhang <jizhan@fb.com>
[ROCm/composable_kernel commit: 754adc70e3]
25 lines
1.2 KiB
CMake
25 lines
1.2 KiB
CMake
add_custom_target(example_batched_gemm_xdl)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_fp32 batched_gemm_xdl_fp32.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_fp32)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_fp16 batched_gemm_xdl_fp16.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_fp16)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_bf16 batched_gemm_xdl_bf16.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_bf16)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_bf16_v3 batched_gemm_xdl_bf16_v3.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_bf16_v3)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_fp8_rowwise_v3 batched_gemm_xdl_fp8_rowwise_v3.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_fp8_rowwise_v3)
|
|
|
|
add_example_executable(example_batched_gemm_xdl_int8 batched_gemm_xdl_int8.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_int8)
|
|
|
|
if(USE_BITINT_EXTENSION_INT4)
|
|
add_example_executable(example_batched_gemm_xdl_int4 batched_gemm_xdl_int4.cpp)
|
|
add_example_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_int4)
|
|
endif()
|