Method Array

src/nvcv/src/priv/Array.cpp:127–158

}

Array::Array(NVCVArrayRequirements reqs, IAllocator &alloc, NVCVResourceType target)
    : m_alloc{alloc}
    , m_reqs{std::move(reqs)}
    , m_target{target}
{
    // Assuming reqs are already validated during its creation

    int64_t bufSize;
    switch (m_target)
    {
    case NVCV_RESOURCE_MEM_CUDA:
        bufSize     = CalcTotalSizeBytes(m_reqs.mem.cudaMem);
        m_memBuffer = m_alloc->allocCudaMem(bufSize, m_reqs.alignBytes);
        break;
    case NVCV_RESOURCE_MEM_HOST:
        bufSize     = CalcTotalSizeBytes(m_reqs.mem.hostMem);
        m_memBuffer = m_alloc->allocHostMem(bufSize, m_reqs.alignBytes);
        break;
    case NVCV_RESOURCE_MEM_HOST_PINNED:
        bufSize     = CalcTotalSizeBytes(m_reqs.mem.hostPinnedMem);
        m_memBuffer = m_alloc->allocHostPinnedMem(bufSize, m_reqs.alignBytes);
        break;
    default:
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Unknown Resource type " << m_target;
        break;
    }

    NVCV_ASSERT(m_memBuffer != nullptr);

    this->exportData(m_data);
    m_data.length = 0;
}

Array::~Array()
{