| 35 | } |
| 36 | |
| 37 | std::pair<TensorLayoutArray, std::unique_ptr<Convolution3DForward>> prepare_sub_opr( |
| 38 | const Convolution3DForwardImpl::AlgoBase::SizeArgs& args) { |
| 39 | auto conv3d_opr = args.handle->create_operator<Convolution3D>(); |
| 40 | set_execution_policy<Convolution3DForward, Convolution3DForward*>( |
| 41 | args.opr, conv3d_opr.get()); |
| 42 | auto&& config = sub_opr_config(args); |
| 43 | conv3d_opr->param() = config.second; |
| 44 | |
| 45 | return {config.first, std::move(conv3d_opr)}; |
| 46 | } |
| 47 | } // namespace |
| 48 | |
| 49 | std::vector<Algorithm::SearchItem> Convolution3DForwardImpl::AlgoGroupConvGeneral:: |
no test coverage detected