Method make_bundle

dnn/src/cambricon/powc/opr_impl.cpp:12–27

using namespace cambricon;

WorkspaceBundle PowCImpl::make_bundle(
        const TensorLayout& src, const TensorLayout& dst) {
    auto handle = concrete_handle(this->handle());
    CnnlTensorDescriptor desc_inp, desc_exp, desc_out;
    auto layout = CNNL_LAYOUT_ARRAY;
    desc_inp.set(src, layout);
    desc_out.set(dst, layout);
    std::vector<size_t> dims = {1};
    desc_exp.set(1, dims, convert_to_cnnl_datatype(src.dtype.enumv()), layout);
    size_t powc_workspace = 0;
    cnnl_check(cnnlGetPowWorkspaceSize(
            handle->cnnl_handle(), desc_inp.desc(), desc_exp.desc(), desc_out.desc(),
            &powc_workspace));
    size_t exp_workspace = sizeof(float);
    return {nullptr, {exp_workspace, powc_workspace}, handle->alignment_requirement()};
}

size_t PowCImpl::get_workspace_in_bytes(
        const TensorLayout& src, const TensorLayout& dst) {