"megatron/vscode:/vscode.git/clone" did not exist on "6529003325ebd5c19e6aa62ef2df4fdf2676c8f4"
CMakeLists.txt 537 Bytes
Newer Older
1
2
3
4
add_custom_target(test_softmax)

add_gtest_executable(test_softmax_fp32 test_softmax_fp32.cpp)
add_gtest_executable(test_softmax_fp16 test_softmax_fp16.cpp)
5
add_gtest_executable(test_softmax_int8 test_softmax_int8.cpp)
6
7
8
target_link_libraries(test_softmax_fp32 PRIVATE utility)
target_link_libraries(test_softmax_fp16 PRIVATE utility)
target_link_libraries(test_softmax_int8 PRIVATE utility)
9
add_dependencies(test_softmax test_softmax_fp32)
10
add_dependencies(test_softmax test_softmax_fp16)
11
add_dependencies(test_softmax test_softmax_int8)