mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-17 11:30:02 +00:00
fe367bc917
* Sync the naming
* Sync the test of layernorm with groupnorm
* Sync the naming
* Minor change for comment and log
* [What] Add saveMean and SaveInvVariance in the interface.
[Why] These can optimize the backward
[ROCm/composable_kernel commit: d4d1147f0a]
18 lines
886 B
CMake
18 lines
886 B
CMake
add_custom_target(test_layernorm)
|
|
|
|
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)
|
|
add_gtest_executable(test_groupnorm_fp32 test_groupnorm_fp32.cpp)
|
|
|
|
target_link_libraries(test_layernorm2d_fp32 PRIVATE utility device_normalization_instance)
|
|
target_link_libraries(test_layernorm2d_fp16 PRIVATE utility device_normalization_instance)
|
|
target_link_libraries(test_groupnorm_fp16 PRIVATE utility device_normalization_instance)
|
|
target_link_libraries(test_groupnorm_fp32 PRIVATE utility device_normalization_instance)
|
|
|
|
add_dependencies(test_layernorm test_layernorm2d_fp32)
|
|
add_dependencies(test_layernorm test_layernorm2d_fp16)
|
|
add_dependencies(test_layernorm test_groupnorm_fp16)
|
|
add_dependencies(test_layernorm test_groupnorm_fp32)
|
|
|