Function make_bundle

dnn/src/cambricon/elemwise/opr_impl.cpp:668–674

namespace {

WorkspaceBundle make_bundle(
        HandleImpl* handle, const SmallVector<size_t>& sizes_in_bytes) {
    WorkspaceBundle wk_bundle{nullptr, sizes_in_bytes, handle->alignment_requirement()};
    void* wk_ptr = handle->alloc(wk_bundle.total_size_in_bytes());
    wk_bundle.set(wk_ptr);
    return wk_bundle;
}

WorkspaceBundle get_unary_ws(
        HandleImpl* handle, const TensorLayoutArray& src, const TensorLayout& dst,