MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / exec

Method exec

dnn/src/rocm/convolution/backward_data/miopen.cpp:78–94  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

76}
77
78void ConvolutionBackwardDataImpl::AlgoMIOpen::exec(const ExecArgs& args) const {
79 MIOpenBwdDataDescs D;
80 args.init_desc(D);
81 auto algo =
82 const_cast<ConvolutionBackwardDataImpl::AlgoMIOpen*>(this)->find_best_algo(
83 args);
84 float alpha = 1.0f, beta = 0.0f;
85 auto status = miopenConvolutionBackwardData(
86 args.handle->miopen_handle(), &alpha, D.diff_desc.desc,
87 args.diff_tensor->raw_ptr(), D.filter_desc.desc,
88 args.filter_tensor->raw_ptr(), D.conv_desc.desc, algo, &beta,
89 D.grad_desc.desc, args.grad_tensor->raw_ptr(), args.workspace.raw_ptr,
90 args.workspace.size);
91 megdnn_assert(
92 status == miopenStatusSuccess, "conv bwd_data failed: %s; info: %s",
93 miopenGetErrorString(status), args.to_string().c_str());
94}
95
96void ConvolutionBackwardDataImpl::AlgoPack::fill_miopen_algos() {}
97

Callers

nothing calls this directly

Calls 5

miopen_handleMethod · 0.80
init_descMethod · 0.45
find_best_algoMethod · 0.45
raw_ptrMethod · 0.45
to_stringMethod · 0.45

Tested by

no test coverage detected