FFNvDecoder/cuda_kernels.h

#pragma once
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <cuda.h>
typedef enum
{
	ITU_601 = 1,
	ITU_709 = 2
} FF_ColorSpace;
namespace cuda_common
{
	cudaError_t setColorSpace(FF_ColorSpace CSC, float hue);
	cudaError_t NV12ToRGBnot(CUdeviceptr d_srcNV12, size_t nSourcePitch, unsigned char* d_dstRGB, int width, int height);
	cudaError_t CUDAToBGR(CUdeviceptr dataY, CUdeviceptr dataUV, size_t pitchY, size_t pitchUV, unsigned char* d_dstRGB, int width, int height);
}