Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 637 Bytes
Newer Older
Anthony Chang's avatar
Standalone softmax kernel (#284)  
Anthony Chang committed
1
2 
add_custom_target(test_softmax)
Adam Osewski's avatar
Softmax unit-test reduction across all and non innermost dims cases. (#406)  
Adam Osewski committed
3
4
5
6
7
8
9
10
11 
add_gtest_executable(test_softmax_rank3 test_softmax_rank3.cpp)
add_gtest_executable(test_softmax_rank4 test_softmax_rank4.cpp)
add_gtest_executable(test_softmax_interface test_softmax_interface.cpp)
target_link_libraries(test_softmax_rank3 PRIVATE utility device_softmax_instance)
target_link_libraries(test_softmax_rank4 PRIVATE utility device_softmax_instance)
target_link_libraries(test_softmax_interface PRIVATE utility device_softmax_instance)
add_dependencies(test_softmax test_softmax_rank3)
add_dependencies(test_softmax test_softmax_rank4)
add_dependencies(test_softmax test_softmax_interface)