"src/targets/gpu/jit/scatter.hpp" did not exist on "150d6d20b0dbb8bd76fa081164d7d99822d6aa63"
CMakeLists.txt 548 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
target_link_libraries(test_softmax_fp32 PRIVATE host_tensor)
target_link_libraries(test_softmax_fp16 PRIVATE host_tensor)
8
target_link_libraries(test_softmax_int8 PRIVATE host_tensor)
9
add_dependencies(test_softmax test_softmax_fp32)
10
11
add_dependencies(test_softmax test_softmax_fp16)
add_dependencies(test_softmax test_softmax_int8)