Method exec

dnn/src/cambricon/convolution/forward/algo.cpp:122–132

}

void ConvolutionForwardImpl::AlgoDefault::exec(const ExecArgs& args) const {
    auto config = prepare_sub_opr(args);
    auto bundle = get_workspace_bundle(config, args);
    bundle.set(args.workspace.raw_ptr);
    auto conv_workspace = bundle.get_workspace(1);
    cnrt_check(cnrtMemsetAsync(
            bundle.get(0), 0, bundle.get_size(0), cnrt_queue(args.opr->handle())));
    config.second->exec(
            args.tensor_src, args.tensor_filter, {bundle.get(0), config.first[0]},
            {nullptr, config.first[1]}, args.tensor_dst, nullptr, conv_workspace);
}

// vim: syntax=cpp.doxygen