mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 13:41:24 +00:00
e823d518cb
* add DeviceGemmXdl * update script * fix naming issue * fix comment * output HostTensorDescriptor * rename * padded GEMM for fwd v4r4r4 nhwc * refactor * refactor * refactor * adding ckProfiler * adding ckProfiler * refactor * fix tuning parameter bug * add more gemm instances * add more fp16 GEMM instances * fix profiler driver * fix bug in tuning parameter * add fp32 gemm instances * small fix * refactor * rename * refactor gemm profiler; adding DeviceConv and conv profiler * refactor * fix * add conv profiler * refactor * adding more GEMM and Conv instance * Create README.md Add build instruction for ckProfiler * Create README.md Add Readme for gemm_xdl example * Update README.md Remove build instruction from top most folder * Update README.md * clean up
43 lines
1.1 KiB
C++
43 lines
1.1 KiB
C++
#ifndef DEVICE_CONV_INSTANTCE_HPP
|
|
#define DEVICE_CONV_INSTANTCE_HPP
|
|
|
|
#include "device_conv.hpp"
|
|
|
|
namespace ck {
|
|
namespace tensor_operation {
|
|
namespace device {
|
|
namespace device_conv_instance {
|
|
|
|
template <ck::index_t NDimSpatial,
|
|
typename InDataType,
|
|
typename WeiDataType,
|
|
typename OutDataType,
|
|
typename InLayout,
|
|
typename WeiLayout,
|
|
typename OutLayout>
|
|
void add_device_conv_fwd_instance(std::vector<DeviceConvFwdPtr>&);
|
|
|
|
template <ck::index_t NDimSpatial,
|
|
typename InDataType,
|
|
typename WeiDataType,
|
|
typename OutDataType,
|
|
typename InLayout,
|
|
typename WeiLayout,
|
|
typename OutLayout>
|
|
void add_device_conv_bwd_instance(std::vector<DeviceConvBwdPtr>&);
|
|
|
|
template <ck::index_t NDimSpatial,
|
|
typename InDataType,
|
|
typename WeiDataType,
|
|
typename OutDataType,
|
|
typename InLayout,
|
|
typename WeiLayout,
|
|
typename OutLayout>
|
|
void add_device_conv_wrw_instance(std::vector<DeviceConvWrwPtr>&);
|
|
|
|
} // namespace device_conv_instance
|
|
} // namespace device
|
|
} // namespace tensor_operation
|
|
} // namespace ck
|
|
#endif
|