mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-18 03:49:41 +00:00
1dcaa3991f
* Fix bug of profiler for layernorm
* 1. Rename layernorm into normalization
2. Decouple softmax from normalization
* clang-format
[ROCm/composable_kernel commit: 1b62bfaa2a]
18 lines
826 B
CMake
18 lines
826 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)
|
|
target_link_libraries(test_layernorm2d_fp16 PRIVATE utility)
|
|
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)
|
|
|