主页 / 问题 / 1607999
Accepted
Stannum chemistry
Asked:2025-03-03 05:19:59 +0000 UTC2025-03-03 05:19:59 +0000 UTC

cudaSetDevice 不起作用，数组返回未初始化的值

  • 772

最近开始学习 cuda c，了解了 cudaSetDevice 函数并决定测试它

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <iostream>
#include <stdio.h>


const int choses = 1;

__global__ void func(int* a,const int choses = 0)
{
    a[blockIdx.x + choses] = 1;
}





int main()
{

    int* a;
    int* a_cpu = new int[2];
    cudaMalloc((void**)&a, 2 * sizeof(int));

    while (true)
    {
        cudaSetDevice(0);
        func <<<1, 1 >>> (a);
        
        cudaSetDevice(1);
        func <<<1, 1 >>> (a, choses);

        cudaMemcpy(a_cpu, a, 2 * sizeof(int), cudaMemcpyDeviceToHost);
        std::cout << a_cpu[0] << "  ";
        std::cout << a_cpu[1] << "\n";
    }
}

该程序采用 2 个数字组成的数组并不断对它们求和，每个元素由其自己的视频卡求和，然后将数据复制到 CPU 变量中，但由于某种原因，程序报告数组元素未初始化

在此处输入图片描述

PS 没有 cudaSetDevice() 一切正常。

c++

1 个回答

  1. Best Answer

    Sync 已退出聊天。在一个设备上运行内核（func）后，您可以立即切换到另一个设备并在那里运行内核，因为 GPU 上的操作是异步执行的。

    内核启动后，在每个设备上调用 cudaDeviceSynchronize 等待执行完成：

    #include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <iostream>

const int choses = 1;

__global__ void func(int* a, const int offset = 0)
{
    a[blockIdx.x + offset] = 1;
}

int main()
{
    int* a_dev0;  // Память на устройстве 0
    int* a_dev1;  // Память на устройстве 1
    int* a_cpu = new int[2];  // Память на хосте

    // Выделяем память на каждом устройстве
    cudaSetDevice(0);
    cudaMalloc((void**)&a_dev0, 2 * sizeof(int));
    cudaSetDevice(1);
    cudaMalloc((void**)&a_dev1, 2 * sizeof(int));

    while (true)
    {
        cudaSetDevice(0);
        func<<<1, 1>>>(a_dev0);
        cudaDeviceSynchronize();  // Синхронизируем устройство 0

        cudaSetDevice(1);
        func<<<1, 1>>>(a_dev1, choses);
        cudaDeviceSynchronize();  // Синхронизируем устройство 1

        // Копируем данные с устройств на хост
        cudaSetDevice(0);
        cudaMemcpy(&a_cpu[0], a_dev0, sizeof(int), cudaMemcpyDeviceToHost);
        cudaSetDevice(1);
        cudaMemcpy(&a_cpu[1], a_dev1, sizeof(int), cudaMemcpyDeviceToHost);

        // Выводим результат
        std::cout << a_cpu[0] << "  " << a_cpu[1] << "\n";
    }

    // Освобождаем память
    cudaSetDevice(0);
    cudaFree(a_dev0);
    cudaSetDevice(1);
    cudaFree(a_dev1);
    delete[] a_cpu;

    return 0;
}

    看起来应该可以

    • 1

相关问题