cuda_kernels.h
579 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 setColorSpace(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);
}