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

Method exec

dnn/src/rocm/convolution/backward_filter/miopen.cpp:81–95  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

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