ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-18 03:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

NHWC Conv2d Bwd weight fp16 ckprofiler and test (#166)

Browse Source 
* change backward weight name

* start add bwd weight lib and profiler

* change tuning paramter

* change output info

* add bwd weight test

* change test info

* using conv_util

* change wgt to weight

* add }

* add fp32

[ROCm/composable_kernel commit: 781cacd2e6]
This commit is contained in:
ltqin
2022-04-05 09:32:00 +08:00
committed by GitHub
parent 573f1de6fa
commit f3eb4639a8
19 changed files with 814 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/include/ck/library/reference_tensor_operation/cpu/reference_conv_backward_weight.hpp
View File

@@ -17,7 +17,7 @@ template <typename InDataType,
typename InElementwiseOperation,
typename WeiElementwiseOperation,
typename OutElementwiseOperation>
struct ReferenceConvWrw : public device::BaseOperator
struct ReferenceConvBwdWeight : public device::BaseOperator
{
// Argument
struct Argument : public device::BaseArgument
@@ -62,7 +62,7 @@ struct ReferenceConvWrw : public device::BaseOperator
// Invoker
struct Invoker : public device::BaseInvoker
{
using Argument = ReferenceConvWrw::Argument;
using Argument = ReferenceConvBwdWeight::Argument;
float Run(const Argument& arg)
{
@@ -163,7 +163,7 @@ struct ReferenceConvWrw : public device::BaseOperator
auto str = std::stringstream();
// clang-format off
str << "ReferenceConvFwd"
str << "ReferenceConvBwdWeight"
<< std::endl;
// clang-format on