mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
63af525c06
* Refine the device batchnorm-backward base API templates and data type assignments * Remove duplicated kernel file * Add batchnorm backward instances and external API * Add batchnorm-backward profiler and tests * Add client example which uses batchnorm backward external API * Merge test/batchnorm_fwd and test/batchnorm_bwd into one directory * Loose the threshold for batchnorm-backward check_err()
5 lines
318 B
CMake
5 lines
318 B
CMake
add_executable(client_batchnorm_fwd_nhwc batchnorm_fwd_nhwc.cpp)
|
|
add_executable(client_batchnorm_bwd_nhwc batchnorm_bwd_nhwc.cpp)
|
|
target_link_libraries(client_batchnorm_fwd_nhwc PRIVATE composable_kernel::device_operations)
|
|
target_link_libraries(client_batchnorm_bwd_nhwc PRIVATE composable_kernel::device_operations)
|