Skip to content

GitLab

"megatron/legacy/model/classification.py" did not exist on "caa9dca52981edc30b8c4930b7f2ace95a531f36"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 703 Bytes
Newer Older
Chao Liu's avatar
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)  
Chao Liu committed
1
2
3 
add_test_executable(test_gemm_fp32 gemm_fp32.cpp)
target_link_libraries(test_gemm_fp32 PRIVATE host_tensor)
target_link_libraries(test_gemm_fp32 PRIVATE device_gemm_instance)
Chao Liu's avatar
Reorganize files, Part 1 (#119)  
Chao Liu committed
4
Chao Liu's avatar
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)  
Chao Liu committed
5
6
7 
add_test_executable(test_gemm_fp16 gemm_fp16.cpp)
target_link_libraries(test_gemm_fp16 PRIVATE host_tensor)
target_link_libraries(test_gemm_fp16 PRIVATE device_gemm_instance)
rocking5566's avatar
Gemm_c_shuffle (4 layouts) X (fp32 bf16 int8) (#131)  
rocking5566 committed
8
Chao Liu's avatar
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)  
Chao Liu committed
9
10
11 
add_test_executable(test_gemm_bf16 gemm_bf16.cpp)
target_link_libraries(test_gemm_bf16 PRIVATE host_tensor)
target_link_libraries(test_gemm_bf16 PRIVATE device_gemm_instance)
Chao Liu's avatar
Reorganize files, Part 1 (#119)  
Chao Liu committed
12
Chao Liu's avatar
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)  
Chao Liu committed
13
14
15 
add_test_executable(test_gemm_int8 gemm_int8.cpp)
target_link_libraries(test_gemm_int8 PRIVATE host_tensor)
target_link_libraries(test_gemm_int8 PRIVATE device_gemm_instance)