CMakeLists.txt 1.39 KB
Newer Older
1
if(DTYPES MATCHES "fp32" OR NOT DEFINED DTYPES)
2
add_test_executable(test_gemm_fp32 gemm_fp32.cpp)
3
target_link_libraries(test_gemm_fp32 PRIVATE utility)
4
target_link_libraries(test_gemm_fp32 PRIVATE device_gemm_instance)
5
6
endif()
if(DTYPES MATCHES "fp16" OR NOT DEFINED DTYPES)
7
add_test_executable(test_gemm_fp16 gemm_fp16.cpp)
8
target_link_libraries(test_gemm_fp16 PRIVATE utility)
9
target_link_libraries(test_gemm_fp16 PRIVATE device_gemm_instance)
10
11
12
13
add_library(gemm_standalone_xdl_fp16_instances STATIC
    instance/gemm_f16_nn_instance.cpp
    instance/gemm_f16_nt_instance.cpp
    instance/gemm_f16_tn_instance.cpp
14
    instance/gemm_wavelet_f16_tn_instance.cpp
15
16
17
18
19
    instance/gemm_f16_tt_instance.cpp
)
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/)
20
21
22
23
24
25
26
27
28
29
30
endif()
if(DTYPES MATCHES "bf16" OR NOT DEFINED DTYPES)
add_test_executable(test_gemm_bf16 gemm_bf16.cpp)
target_link_libraries(test_gemm_bf16 PRIVATE utility)
target_link_libraries(test_gemm_bf16 PRIVATE device_gemm_instance)
endif()
if(DTYPES MATCHES "int8" OR NOT DEFINED DTYPES)
add_test_executable(test_gemm_int8 gemm_int8.cpp)
target_link_libraries(test_gemm_int8 PRIVATE utility)
target_link_libraries(test_gemm_int8 PRIVATE device_gemm_instance)
endif()