Function ValidateTensorBufferStrided

src/nvcv/src/priv/TensorWrapDataStrided.cpp:35–105

namespace nvcv::priv {

static void ValidateTensorBufferStrided(const NVCVTensorData &tdata)
{
    NVCV_ASSERT(tdata.bufferType == NVCV_TENSOR_BUFFER_STRIDED_CUDA);

    const NVCVTensorBufferStrided &buffer = tdata.buffer.strided;

    if (buffer.basePtr == nullptr)
    {
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Memory buffer must not be NULL";
    }

    int rank = tdata.rank;

    if (rank <= 0)
    {
        throw Exception(NVCV_ERROR_INVALID_ARGUMENT, "Number of dimensions must be >= 1, not %d", rank);
    }

    for (int i = 0; i < rank; ++i)
    {
        if (tdata.shape[i] < 1)
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT) << "Shape #" << i << " must be >= 1, not " << tdata.shape[i];
        }
    }

    DataType dtype{tdata.dtype};

    int firstPacked = IsChannelLast(tdata.layout) ? std::max(0, rank - 2) : rank - 1;

    int prevStrideConsidered = buffer.strides[rank - 1];
    int prevSizeConsidered   = tdata.shape[rank - 1];

    // Test packed dimensions
    int dim;
    for (dim = rank - 1; dim >= firstPacked; --dim)
    {
        if (tdata.shape[dim] <= 1)
        {
            continue;
        }

        int correctPitch = dim == rank - 1 ? dtype.strideBytes() : prevStrideConsidered * prevSizeConsidered;
        if (buffer.strides[dim] != correctPitch)
        {
            throw Exception(NVCV_ERROR_INVALID_ARGUMENT)
                << "Pitch of dimension " << dim << " must be == " << correctPitch << " (packed)"
                << ", but it is " << buffer.strides[dim];
        }
        prevStrideConsidered = buffer.strides[dim];
        prevSizeConsidered   = tdata.shape[dim];
    }

    // Test non-packed dimensions
    for (; dim >= 0; --dim)
    {
        if (tdata.shape[dim] <= 1)
        {