CUDA：实现归并排序(附完整源码)

源代码大师

于 2024-03-02 13:20:10 发布

阅读量320

点赞数

分类专栏： CUDA实战教程文章标签：算法 CUDA

不予转载，严禁转载，违者必纠。

本文链接：https://blog.csdn.net/it_xiangqiang/article/details/136414034

版权

CUDA实战教程专栏收录该内容

246 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文展示了一个使用CUDA进行归并排序的实例，通过在主机生成随机数据，将其传输到设备，利用CUDA核函数进行排序，再将结果返回主机显示。提供完整源码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

CUDA：实现归并排序

以下是一个使用CUDA实现归并排序的示例代码：

#include <iostream>
#include <cuda_runtime_api.h>

#define BLOCK_SIZE 256

// CUDA核函数：归并排序
__global__ void mergeSort(int* data, int* temp, int left, int right) {
    int start = left + blockIdx.x * blockDim.x + threadIdx.x;
    int end = min(start + blockDim.x, right);

    if (start < right) {
        // 拷贝数据到临时数组
        temp[start] = data[start];
        __syncthreads();

        // 归并排序
        for (int stride = 1; stride < blockDim.x; stride *= 2) {
            int index = 2 * stride * threadIdx.x;
            if (index < blockDim.x) {
                int left = start + index;
                int right = min(start + 2 * stride, end);