InfiniTensor/include/cuda/resize.cuh

#pragma once
#include "cuda/cuda_common.h"

typedef struct {
    int nDims;
    int oDims[4];
    int inDims[4];
    int inStride[4];
    float scale[4];
    float roiS[4];
    float roiE[4];
} MetaData;

namespace infini {
void resize_kernel_nearest(float *in, float *out, const MetaData &metaData,
                           size_t num, int coordinateMode, int nearestMode);
void resize_kernel_linear(float *in, float *out, const MetaData &metaData,
                          size_t num, int coordinateMode);
void resize_kernel_cubic(float *in, float *out, const MetaData &metaData,
                         size_t num, int coordinateMode);
} // namespace infini