Fix gemm splitk instance build

65f38a67 · Rostyslav Geyyer · ce12637a · 65f38a67
Commit 65f38a67 authored Oct 06, 2023 by Rostyslav Geyyer
Show whitespace changes
Inline Side-by-side

Showing with 13 additions and 9 deletions

library/src/tensor_operation_instance/gpu/gemm_splitk/CMakeLists.txt .../tensor_operation_instance/gpu/gemm_splitk/CMakeLists.txt +13 -9

No files found.
--- a/library/src/tensor_operation_instance/gpu/gemm_splitk/CMakeLists.txt
+++ b/library/src/tensor_operation_instance/gpu/gemm_splitk/CMakeLists.txt
@@ -7,7 +7,10 @@ list(APPEND GEMM_SPLITK_INSTANCES device_gemm_xdl_splitk_f32_f32_f32_mk_kn_mn_in
                                  device_gemm_xdl_splitk_f16_f16_f16_mk_kn_mn_instance.cpp
                                  device_gemm_xdl_splitk_f16_f16_f16_mk_nk_mn_instance.cpp
                                  device_gemm_xdl_splitk_f16_f16_f16_km_kn_mn_instance.cpp
-                                  device_gemm_xdl_splitk_f16_f16_f16_km_nk_mn_instance.cpp
+                                  device_gemm_xdl_splitk_f16_f16_f16_km_nk_mn_instance.cpp)
+
+if((DTYPES MATCHES "fp8" AND DTYPES MATCHES "fp16") OR NOT DEFINED DTYPES)
+    list(APPEND GEMM_SPLITK_INSTANCES
        device_gemm_xdl_splitk_fp8_f16_f16_mk_kn_mn_instance.cpp
        device_gemm_xdl_splitk_fp8_f16_f16_mk_nk_mn_instance.cpp
        device_gemm_xdl_splitk_fp8_f16_f16_km_kn_mn_instance.cpp
@@ -16,5 +19,6 @@ list(APPEND GEMM_SPLITK_INSTANCES device_gemm_xdl_splitk_f32_f32_f32_mk_kn_mn_in
        device_gemm_xdl_splitk_f16_fp8_f16_mk_nk_mn_instance.cpp
        device_gemm_xdl_splitk_f16_fp8_f16_km_kn_mn_instance.cpp
        device_gemm_xdl_splitk_f16_fp8_f16_km_nk_mn_instance.cpp)
+endif()

 add_instance_library(device_gemm_splitk_instance ${GEMM_SPLITK_INSTANCES})