CMakeLists.txt 827 Bytes
Newer Older
rocking5566's avatar
rocking5566 committed
1
2
add_custom_target(test_layernorm)

rocking5566's avatar
rocking5566 committed
3
4
5
add_gtest_executable(test_layernorm2d_fp32 test_layernorm2d_fp32.cpp)
add_gtest_executable(test_layernorm2d_fp16 test_layernorm2d_fp16.cpp)
add_gtest_executable(test_groupnorm_fp16 test_groupnorm_fp16.cpp)
guangzlu's avatar
guangzlu committed
6
add_gtest_executable(test_groupnorm_fp32 test_groupnorm_fp32.cpp) 
7

guangzlu's avatar
guangzlu committed
8
target_link_libraries(test_layernorm2d_fp32 PRIVATE utility) 
rocking5566's avatar
rocking5566 committed
9
10
11
12
13
14
15
16
target_link_libraries(test_layernorm2d_fp16 PRIVATE utility)
target_link_libraries(test_groupnorm_fp16 PRIVATE utility device_normalization_instance)
target_link_libraries(test_groupnorm_fp32 PRIVATE utility device_normalization_instance)

add_dependencies(test_layernorm test_layernorm2d_fp32)
add_dependencies(test_layernorm test_layernorm2d_fp16)
add_dependencies(test_layernorm test_groupnorm_fp16)
add_dependencies(test_layernorm test_groupnorm_fp32)