Skip to content

GitLab

"megatron/vscode:/vscode.git/clone" did not exist on "6529003325ebd5c19e6aa62ef2df4fdf2676c8f4"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 537 Bytes
Newer Older
Anthony Chang's avatar
Standalone softmax kernel (#284)  
Anthony Chang committed
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)
Anthony Chang's avatar
Standalone sweep once softmax kernel w/ ckProfiler (#295)  
Anthony Chang committed
5 
add_gtest_executable(test_softmax_int8 test_softmax_int8.cpp)
Chao Liu's avatar
Clean up conv example, Instances, profiler and test (#324)  
Chao Liu committed
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)
Anthony Chang's avatar
Standalone softmax kernel (#284)  
Anthony Chang committed
9 
add_dependencies(test_softmax test_softmax_fp32)
Anthony Chang's avatar
Standalone sweep once softmax kernel w/ ckProfiler (#295)  
Anthony Chang committed
10 
add_dependencies(test_softmax test_softmax_fp16)
Chao Liu's avatar
Clean up conv example, Instances, profiler and test (#324)  
Chao Liu committed
11 
add_dependencies(test_softmax test_softmax_int8)