Method do_exec_ct

dnn/src/cambricon/powc/opr_impl.cpp:35–56

template <typename T>
void PowCImpl::do_exec_ct(
        _megdnn_tensor_in src, _megdnn_tensor_out dst, _megdnn_workspace workspace) {
    auto handle = concrete_handle(this->handle());
    auto ws_bundle = make_bundle(src.layout, dst.layout);
    ws_bundle.set(workspace.raw_ptr);

    CnnlTensorDescriptor desc_inp, desc_exp, desc_out;
    auto cnnl_layout = CNNL_LAYOUT_ARRAY;
    desc_inp.set(src.layout, cnnl_layout);
    desc_out.set(dst.layout, cnnl_layout);
    std::vector<size_t> dims = {1};
    desc_exp.set(1, dims, convert_to_cnnl_datatype(DTypeTrait<T>::enumv), cnnl_layout);
    T v_exp = static_cast<T>(param().exp);
    T* dev_exp = static_cast<T*>(ws_bundle.get(0));
    cnnl_check(cnnlFill_v3(
            handle->cnnl_handle(), CNNL_POINTER_MODE_HOST, (void*)&v_exp,
            desc_exp.desc(), dev_exp));
    cnnl_check(cnnlPow(
            handle->cnnl_handle(), CNNL_COMPUTATION_HIGH_PRECISION, desc_inp.desc(),
            src.raw_ptr(), desc_exp.desc(), dev_exp, ws_bundle.get(1),
            ws_bundle.get_size(1), desc_out.desc(), dst.raw_ptr()));
}

void PowCImpl::exec(
        _megdnn_tensor_in src, _megdnn_tensor_out dst, _megdnn_workspace workspace) {