Method Wrap

python/mod_cvcuda/nvcv/Array.cpp:119–139

} // namespace

std::shared_ptr<Array> Array::Wrap(ExternalBuffer &buffer)
{
    const DLTensor &dlTensor = buffer.dlTensor();

    nvcv::ArrayDataCuda data{FillNVCVArrayDataCUDA(dlTensor)};

    // This is the key of a tensor wrapper.
    // All tensor wrappers have the same key.
    Array::Key key;
    // We take this opportunity to remove from cache all wrappers that aren't
    // being used. They aren't reusable anyway.
    Cache::Instance().removeAllNotInUseMatching(key);

    auto array = std::shared_ptr<Array>(new Array(data, py::cast(buffer.shared_from_this())));

    // Need to add wrappers to cache so that they don't get destroyed by
    // the cuda stream when they're last used, and python script isn't
    // holding a reference to them. If we don't do it, things might break.
    Cache::Instance().add(*array);
    return array;
}

std::shared_ptr<Array> Array::ResizeArray(Array &array, int64_t length)
{