mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-18 03:49:41 +00:00
b64f30e733
* Rename folder
* Add layernorm 4d fwd example
* Rename original layernorm example
* Add layernorm 4d f16 test
* Add layernorm4d_fwd client example
* Support layernorm4D in ckProfiler
* Rename groupnorm to groupnorm fwd in example
* Rename layernorm and group fwd in test
* Rename normalization to normalization_fwd (instances)
* Add fwd to DeviceNormalization
* Rename external api header
* Rename folder, because we can also add bwd in this folder
* Add fwd in layernorm and groupnorm (profiler
* Fix compile error
---------
Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
[ROCm/composable_kernel commit: a3d9a2cd42]
31 lines
1.4 KiB
CMake
31 lines
1.4 KiB
CMake
add_custom_target(test_normalization_fwd)
|
|
add_gtest_executable(test_layernorm2d_fwd_fp32 test_layernorm2d_fwd_fp32.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_layernorm2d_fwd_fp32 PRIVATE utility device_normalization_fwd_instance)
|
|
add_dependencies(test_normalization_fwd test_layernorm2d_fwd_fp32)
|
|
endif()
|
|
|
|
add_gtest_executable(test_groupnorm_fwd_fp32 test_groupnorm_fwd_fp32.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_groupnorm_fwd_fp32 PRIVATE utility device_normalization_fwd_instance)
|
|
add_dependencies(test_normalization_fwd test_groupnorm_fwd_fp32)
|
|
endif()
|
|
|
|
add_gtest_executable(test_layernorm2d_fwd_fp16 test_layernorm2d_fwd_fp16.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_layernorm2d_fwd_fp16 PRIVATE utility device_normalization_fwd_instance)
|
|
add_dependencies(test_normalization_fwd test_layernorm2d_fwd_fp16)
|
|
endif()
|
|
|
|
add_gtest_executable(test_layernorm4d_fwd_fp16 test_layernorm4d_fwd_fp16.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_layernorm4d_fwd_fp16 PRIVATE utility device_normalization_fwd_instance)
|
|
add_dependencies(test_normalization_fwd test_layernorm4d_fwd_fp16)
|
|
endif()
|
|
|
|
add_gtest_executable(test_groupnorm_fwd_fp16 test_groupnorm_fwd_fp16.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_groupnorm_fwd_fp16 PRIVATE utility device_normalization_fwd_instance)
|
|
add_dependencies(test_normalization_fwd test_groupnorm_fwd_fp16)
|
|
endif()
|