Fix gemm multiply add instnace build

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

Showing with 6 additions and 3 deletions

library/src/tensor_operation_instance/gpu/gemm_multiply_add/CMakeLists.txt ...r_operation_instance/gpu/gemm_multiply_add/CMakeLists.txt +6 -3

No files found.
--- a/library/src/tensor_operation_instance/gpu/gemm_multiply_add/CMakeLists.txt
+++ b/library/src/tensor_operation_instance/gpu/gemm_multiply_add/CMakeLists.txt
 set(GEMM_MULTIPLY_ADD_INSTANCES)
 list(APPEND GEMM_MULTIPLY_ADD_INSTANCES device_gemm_multiply_add_xdl_c_shuffle_f16_f16_f16_f16_f16_mk_kn_mn_mn_mn_instance.cpp
-                                        device_gemm_multiply_add_xdl_c_shuffle_f16_f16_f16_f16_f16_mk_nk_mn_mn_mn_instance.cpp
+                                        device_gemm_multiply_add_xdl_c_shuffle_f16_f16_f16_f16_f16_mk_nk_mn_mn_mn_instance.cpp)
+if((DTYPES MATCHES "fp8" AND DTYPES MATCHES "fp16" AND DTYPES MATCHES "fp32") OR NOT DEFINED DTYPES)
+    list(APPEND GEMM_MULTIPLY_ADD_INSTANCES
        device_gemm_multiply_add_xdl_c_shuffle_f16_f8_f32_f32_f16_mk_kn_mn_mn_mn_instance.cpp
        device_gemm_multiply_add_xdl_c_shuffle_f16_f8_f32_f32_f16_mk_nk_mn_mn_mn_instance.cpp)
+endif()
 add_instance_library(device_gemm_multiply_add_instance ${GEMM_MULTIPLY_ADD_INSTANCES})