mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 17:26:00 +00:00
a3d9a2cd42
* 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>
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()
|