CMakeLists.txt 1.15 KB
Newer Older
1
add_test_executable(test_gemm_fp32 gemm_fp32.cpp)
2
3
if(result EQUAL 0)
    target_link_libraries(test_gemm_fp32 PRIVATE utility device_gemm_instance)
4
endif()
5
add_test_executable(test_gemm_fp16 gemm_fp16.cpp)
6
7
8
if(result EQUAL 0)
    target_link_libraries(test_gemm_fp16 PRIVATE utility device_gemm_instance)
    add_library(gemm_standalone_xdl_fp16_instances STATIC
9
10
11
    instance/gemm_f16_nn_instance.cpp
    instance/gemm_f16_nt_instance.cpp
    instance/gemm_f16_tn_instance.cpp
12
    instance/gemm_wavelet_f16_tn_instance.cpp
13
    instance/gemm_f16_tt_instance.cpp
14
15
    )
endif()
16
add_test_executable(test_gemm_standalone_xdl_fp16 gemm_standalone_xdl_fp16.cpp)
17
18
19
if(result EQUAL 0)
    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
endif()
add_test_executable(test_gemm_bf16 gemm_bf16.cpp)
22
23
if(result EQUAL 0)
    target_link_libraries(test_gemm_bf16 PRIVATE utility device_gemm_instance)
24
25
endif()
add_test_executable(test_gemm_int8 gemm_int8.cpp)
26
27
if(result EQUAL 0)
    target_link_libraries(test_gemm_int8 PRIVATE utility device_gemm_instance)
28
endif()