Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "1e731018545cd1a11a266ed356db975db870adb4"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 821 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 
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)
Anthony Chang's avatar
host softmax can run with pre-calculated stats for debug purposes  
Anthony Chang committed
6 
add_gtest_executable(test_softmax_host_ref test_softmax_host_ref.cpp)
Adam Osewski's avatar
Softmax unit-test reduction across all and non innermost dims cases. (#406)  
Adam Osewski committed
7
8
9 
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)
Anthony Chang's avatar
host softmax can run with pre-calculated stats for debug purposes  
Anthony Chang committed
10 
target_link_libraries(test_softmax_host_ref PRIVATE utility)
Adam Osewski's avatar
Softmax unit-test reduction across all and non innermost dims cases. (#406)  
Adam Osewski committed
11
12
13 
add_dependencies(test_softmax test_softmax_rank3)
add_dependencies(test_softmax test_softmax_rank4)
add_dependencies(test_softmax test_softmax_interface)
Anthony Chang's avatar
host softmax can run with pre-calculated stats for debug purposes  
Anthony Chang committed
14 
add_dependencies(test_softmax test_softmax_host_ref)