Skip to content

GitLab

"test/torchaudio_unittest/models/models_test.py" did not exist on "1f136671b84071a2fe1d5b762df64f3a76310c31"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 905 Bytes
Newer Older
rocking5566's avatar
Gemm+layernorm instance, ckProfiler, client example (#568)  
rocking5566 committed
1 
add_custom_target(test_normalization)
rocking5566's avatar
Standalone layernorm (#315)  
rocking5566 committed
2
rocking5566's avatar
Group norm (#417)  
rocking5566 committed
3
4
5 
add_gtest_executable(test_layernorm2d_fp32 test_layernorm2d_fp32.cpp)
add_gtest_executable(test_layernorm2d_fp16 test_layernorm2d_fp16.cpp)
add_gtest_executable(test_groupnorm_fp16 test_groupnorm_fp16.cpp)
rocking5566's avatar
Gemm+layernorm instance, ckProfiler, client example (#568)  
rocking5566 committed
6 
add_gtest_executable(test_groupnorm_fp32 test_groupnorm_fp32.cpp)
Chao Liu's avatar
Clean up conv example, Instances, profiler and test (#324)  
Chao Liu committed
7
rocking5566's avatar
Refine layernorm naming and test code (#497)  
rocking5566 committed
8
9 
target_link_libraries(test_layernorm2d_fp32 PRIVATE utility device_normalization_instance)
target_link_libraries(test_layernorm2d_fp16 PRIVATE utility device_normalization_instance)
rocking5566's avatar
Group norm (#417)  
rocking5566 committed
10
11
12 
target_link_libraries(test_groupnorm_fp16 PRIVATE utility device_normalization_instance)
target_link_libraries(test_groupnorm_fp32 PRIVATE utility device_normalization_instance)
rocking5566's avatar
Gemm+layernorm instance, ckProfiler, client example (#568)  
rocking5566 committed
13
14
15
16 
add_dependencies(test_normalization test_layernorm2d_fp32)
add_dependencies(test_normalization test_layernorm2d_fp16)
add_dependencies(test_normalization test_groupnorm_fp16)
add_dependencies(test_normalization test_groupnorm_fp32)