cuda四舍五入取整

在CUDA中，可以使用`round()`函数来进行四舍五入取整操作。`round()`函数是一个内置的数学函数，它接受一个浮点数作为输入，并返回最接近该浮点数的整数值。

以下是使用`round()`函数进行四舍五入取整的示例代码：

#include <stdio.h>
#include <math.h>

__global__ void roundKernel(float* input, int* output, int size) {
    int tid = blockIdx.x * blockDim.x + threadIdx.x;
    if (tid < size) {
        output[tid] = round(input[tid]);
    }
}

int main() {
    int size = 10;
    float input[size] = {1.2, 2.7, 3.5, 4.9, 5.1, 6.8, 7.3, 8.6, 9.2, 10.7};
    float* d_input;
    int* d_output;
    cudaMalloc((void**)&d_input, size * sizeof(float));
    cudaMalloc((void**)&d_output, size * sizeof(int));
    cudaMemcpy(d_input, input, size * sizeof(float), cudaMemcpyHostToDevice);
    int blockSize = 256;
    int gridSize = (size + blockSize - 1) / blockSize;
    roundKernel<<<gridSize, blockSize>>>(d_input, d_output, size);
    int output[size];
    cudaMemcpy(output, d_output, size * sizeof(int), cudaMemcpyDeviceToHost);
    for (int i = 0; i < size; i++) {
        printf("%.1f -> %d\n", input[i], output[i]);
    }
    cudaFree(d_input);
    cudaFree(d_output);
    return 0;
}

在上述示例代码中，我们首先定义了一个`roundKernel`函数，该函数接受一个浮点数数组作为输入，并使用`round()`函数对每个元素进行四舍五入取整操作，将结果保存到一个整数数组中。

然后，在主函数中，我们定义了一个大小为10的浮点数数组`input`，并将其拷贝到CUDA设备内存中。接着，我们调用`roundKernel`函数来进行四舍五入取整操作，并将结果拷贝回主机内存。最后，我们遍历整数数组，并打印每个元素的原始值和取整后的值。

希望以上解答对您有帮助！如果您还有其他问题，请随时提问。