Skip to content

GitLab

"src/targets/gpu/jit/scatter.hpp" did not exist on "150d6d20b0dbb8bd76fa081164d7d99822d6aa63"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 548 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)
Anthony Chang's avatar
Standalone softmax kernel (#284)  
Anthony Chang committed
6
7 
target_link_libraries(test_softmax_fp32 PRIVATE host_tensor)
target_link_libraries(test_softmax_fp16 PRIVATE host_tensor)
Anthony Chang's avatar
Standalone sweep once softmax kernel w/ ckProfiler (#295)  
Anthony Chang committed
8 
target_link_libraries(test_softmax_int8 PRIVATE host_tensor)
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
11 
add_dependencies(test_softmax test_softmax_fp16)
add_dependencies(test_softmax test_softmax_int8)