CMakeLists.txt 1.97 KB
Newer Older
1
add_test_executable(test_gemm_fp32 gemm_fp32.cpp)
2
target_link_libraries(test_gemm_fp32 PRIVATE utility)
3
target_link_libraries(test_gemm_fp32 PRIVATE device_gemm_instance)
Chao Liu's avatar
Chao Liu committed
4

5
add_test_executable(test_gemm_fp16 gemm_fp16.cpp)
6
target_link_libraries(test_gemm_fp16 PRIVATE utility)
7
target_link_libraries(test_gemm_fp16 PRIVATE device_gemm_instance)
8

9
add_test_executable(test_gemm_bf16 gemm_bf16.cpp)
10
target_link_libraries(test_gemm_bf16 PRIVATE utility)
11
target_link_libraries(test_gemm_bf16 PRIVATE device_gemm_instance)
Chao Liu's avatar
Chao Liu committed
12

13
add_test_executable(test_gemm_int8 gemm_int8.cpp)
14
target_link_libraries(test_gemm_int8 PRIVATE utility)
15
target_link_libraries(test_gemm_int8 PRIVATE device_gemm_instance)
16
17

add_library(gemm_standalone_xdl_fp16_instances STATIC
Po-Yen, Chen's avatar
Po-Yen, Chen committed
18
19
    # instance/gemm_f16_nn_instance.cpp
    # instance/gemm_f16_nt_instance.cpp
Po-Yen, Chen's avatar
Po-Yen, Chen committed
20
    instance/gemm_f16_tn_instance.cpp
Po-Yen, Chen's avatar
Po-Yen, Chen committed
21
    # instance/gemm_f16_tt_instance.cpp
22
)
23

Po-Yen, Chen's avatar
Po-Yen, Chen committed
24
25
26
27
28
29
30
set_source_files_properties(instance/gemm_f16_tn_instance.cpp PROPERTIES COMPILE_OPTIONS "--save-temps;-Wno-gnu-line-marker")
set_source_files_properties(instance/gemm_f16_tn_instance.cpp PROPERTIES COMPILE_DEFINITIONS "ENABLE_PIPELINE_V2_OPT;GEMM_LAYOUT=LAYOUT_TN;IGLP_OPT_STRATEGY=1")

# set_source_files_properties(instance/gemm_f16_nn_instance.cpp PROPERTIES COMPILE_DEFINITIONS "ENABLE_PIPELINE_V2_OPT;IGLP_OPT_STRATEGY=1")
# set_source_files_properties(instance/gemm_f16_tn_instance.cpp PROPERTIES COMPILE_DEFINITIONS "ENABLE_PIPELINE_V2_OPT;IGLP_OPT_STRATEGY=1")
# set_source_files_properties(instance/gemm_f16_nt_instance.cpp PROPERTIES COMPILE_DEFINITIONS "ENABLE_PIPELINE_V2_OPT;IGLP_OPT_STRATEGY=1")
# set_source_files_properties(instance/gemm_f16_tt_instance.cpp PROPERTIES COMPILE_DEFINITIONS "ENABLE_PIPELINE_V2_OPT;IGLP_OPT_STRATEGY=1")
31

32
33
34
add_test_executable(test_gemm_standalone_xdl_fp16 gemm_standalone_xdl_fp16.cpp)
target_link_libraries(test_gemm_standalone_xdl_fp16 PRIVATE gemm_standalone_xdl_fp16_instances utility)
target_include_directories(test_gemm_standalone_xdl_fp16 PRIVATE instance/)