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

3
4
add_gtest_executable(test_layernorm2d_fp32 test_layernorm2d_fp32.cpp)
add_gtest_executable(test_layernorm2d_fp16 test_layernorm2d_fp16.cpp)
rocking's avatar
rocking committed
5
add_gtest_executable(test_groupnorm_fp16 test_groupnorm_fp16.cpp)
6

7
8
target_link_libraries(test_layernorm2d_fp32 PRIVATE utility)
target_link_libraries(test_layernorm2d_fp16 PRIVATE utility)
rocking's avatar
rocking committed
9
target_link_libraries(test_groupnorm_fp16 PRIVATE utility)
10

11
12
add_dependencies(test_layernorm test_layernorm2d_fp32)
add_dependencies(test_layernorm test_layernorm2d_fp16)
rocking's avatar
rocking committed
13
add_dependencies(test_layernorm test_groupnorm_fp16)