cuda_kernels.h 580 Bytes
#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
{
	ITU601 = 1,
	ITU709 = 2
} e_ColorSpace;

namespace cuda_common
{
	cudaError_t setColorSpace2(e_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);
}