mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-02 12:41:26 +00:00
a1b2441f8d
* File renaming and class renaming for device element-wise operation * Add batchnorm-infer instances, external API and client example * Add batchnorm-infer profiler module and gtests * Remove file device_elementwise_extension.hpp and move NormalizeInInfer operation to element_wise_operation.hpp * Remove the using of class aliasing for DeviceElementwiseForBatchNormInfer * Rename class and file due to conflict from device_elementwise_2d.hpp * Fix namespace in batcnnorm_infer_nhwc client example
7 lines
483 B
CMake
7 lines
483 B
CMake
add_executable(client_batchnorm_fwd_nhwc batchnorm_fwd_nhwc.cpp)
|
|
add_executable(client_batchnorm_bwd_nhwc batchnorm_bwd_nhwc.cpp)
|
|
add_executable(client_batchnorm_infer_nhwc batchnorm_infer_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)
|
|
target_link_libraries(client_batchnorm_infer_nhwc PRIVATE composable_kernel::device_operations)
|