用C++编写的NV12到RGB24转换代码

// 导入必要的库
#include <iostream>
#include <cuda_runtime.h>
#include <cuda_runtime_api.h>
#include <cuda_fp16.h>
#include <cstdio>

// 定义输入和输出纹理
__device__ float4 input_texture[2];
__device__ float4 output_texture[2];

// 定义转换函数
__global__ void convert_kernel(const float* input, float* output, int width, int height) {
    int idx = blockIdx.x * blockDim.x + threadIdx.x;
    if (idx < width * height) {
        int pos = idx * 2;
        output[idx] = input[pos];
        output[idx + 1] = input[pos + 1];
    }
}

int main() {
    // 初始化输入和输出纹理
    cudaMemcpy(input_texture, input_ptr, sizeof(float4), cudaMemcpyDeviceToDevice);
    cudaMemcpy(output_texture, output_ptr, sizeof(float4), cudaMemcpyDeviceToDevice);

    // 设置转换参数
    int width = input_texture[0].x;
    int height = input_texture[0].y;
    int num_threads_per_block = 256;
    int num_blocks = (width * height) / (num_threads_per_block * num_threads_per_block);

    // 调用转换内核
    convert_kernel<<<num_blocks, num_threads_per_block>>>(input_texture, output_texture, width, height);

    // 等待转换完成
    cudaDeviceSynchronize();

    // 输出结果
    float result[2];
    cudaMemcpy(result, output_ptr, sizeof(float), cudaMemcpyDeviceToHost);

    // 打印结果
    printf("R: %f G: %f B: %f
", result[0], result[1]);

    // 释放内存
    cudaFree(input_ptr);
    cudaFree(output_ptr);

    return 0;
}