| 76 | } |
| 77 | |
| 78 | void ConvolutionForwardImpl::AlgoMIOpen::exec(const ExecArgs& args) const { |
| 79 | MIOpenForwardDescs D; |
| 80 | args.init_desc(D); |
| 81 | auto algo = |
| 82 | const_cast<ConvolutionForwardImpl::AlgoMIOpen*>(this)->find_best_algo(args); |
| 83 | float alpha = 1.0f, beta = 0.0f; |
| 84 | auto status = miopenConvolutionForward( |
| 85 | args.handle->miopen_handle(), &alpha, D.src_desc.desc, |
| 86 | args.src_tensor->raw_ptr(), D.filter_desc.desc, |
| 87 | args.filter_tensor->raw_ptr(), D.conv_desc.desc, algo, &beta, |
| 88 | D.dst_desc.desc, args.dst_tensor->raw_ptr(), args.workspace.raw_ptr, |
| 89 | args.workspace.size); |
| 90 | megdnn_assert( |
| 91 | status == miopenStatusSuccess, "conv fwd failed: %s; info: %s", |
| 92 | miopenGetErrorString(status), args.to_string().c_str()); |
| 93 | } |
| 94 | |
| 95 | void ConvolutionForwardImpl::AlgoPack::fill_miopen_algos() { |
| 96 | megdnn_throw( |
nothing calls this directly
no test coverage detected