Function ResizeCropConvertReformat

bench/BenchResizeCropConvertReformat.cpp:26–112

template<typename T>
inline void ResizeCropConvertReformat(nvbench::state &state, nvbench::type_list<T>)
try
{
    long3 srcShape = benchutils::GetShape<3>(state.get_string("shape"));
    long  varShape = state.get_int64("varShape");

    NVCVInterpolationType interpType = benchutils::GetInterpolationType(state.get_string("interpolation"));

    int2 cropPos{1, 1};

    NVCVSize2D resize;

    if (state.get_string("resizeType") == "EXPAND")
    {
        resize = NVCVSize2D{(int)(srcShape.y * 2), (int)(srcShape.z * 2)};
    }
    else if (state.get_string("resizeType") == "CONTRACT")
    {
        resize = NVCVSize2D{(int)(srcShape.y / 2), (int)(srcShape.z / 2)};
    }
    else
    {
        throw std::invalid_argument("Invalid resizeType = " + state.get_string("resizeType"));
    }

    NVCVChannelManip manip;

    if (state.get_string("manip") == "NO_OP")
    {
        manip = NVCV_CHANNEL_NO_OP;
    }
    else if (state.get_string("manip") == "REVERSE")
    {
        manip = NVCV_CHANNEL_REVERSE;
    }
    else
    {
        throw std::invalid_argument("Invalid channel manipulation = " + state.get_string("manip"));
    }

    using BT = nvcv::cuda::BaseType<T>;
    long nc  = nvcv::cuda::NumElements<T>;

    long3 dstShape{srcShape.x, resize.h - cropPos.y, resize.w - cropPos.x};

    if (dstShape.y <= 0 || dstShape.z <= 0)
    {
        throw std::invalid_argument("Invalid shape and resizeType");
    }

    state.add_global_memory_reads(srcShape.x * srcShape.y * srcShape.z * sizeof(T));
    state.add_global_memory_writes(dstShape.x * dstShape.y * dstShape.z * sizeof(T));

    cvcuda::ResizeCropConvertReformat op;

    // clang-format off

    if (varShape < 0) // negative var shape means use Tensor