问CUDA钉住从设备中释放的内存
EN

Stack Overflow用户

提问于 2013-05-07 02:59:40

回答 3查看 1.9K关注 0票数 2

CUDA 5，设备功能3.5，VS 2012，64位Win 2012 Server。

线程之间没有共享内存访问，每个线程都是独立的。

我用的是带零拷贝的固定内存。只有当我在主机上发出cudaDeviceSynchronize时，我才能从主机读取设备所写的固定内存。

我希望能够：

设备更新后，立即刷新固定内存。
不阻塞设备线程(可能通过异步复制)

每次写完设备后，我都试着打电话给__threadfence_system和__threadfence，但是没有冲洗。

下面是一个完整的示例数据自动化系统代码，演示了我的问题：

#include <conio.h>
#include <cstdio>
#include "cuda.h"
#include "cuda_runtime.h"
#include "device_launch_parameters.h"

__global__ void Kernel(volatile float* hResult) 
{
    int tid = threadIdx.x + blockIdx.x * blockDim.x;

    printf("Kernel %u: Before Writing in Kernel\n", tid);
    hResult[tid] = tid + 1;
    __threadfence_system();
    // expecting that the data is getting flushed to host here!
    printf("Kernel %u: After Writing in Kernel\n", tid);
    // time waster for-loop (sleep)
    for (int timeWater = 0; timeWater  < 100000000; timeWater++);
}

void main()
{
    size_t blocks = 2;
    volatile float* hResult;
    cudaHostAlloc((void**)&hResult,blocks*sizeof(float),cudaHostAllocMapped);
    Kernel<<<1,blocks>>>(hResult);
    int filledElementsCounter = 0;
    // naiive thread implementation that can be impelemted using 
    // another host thread
    while (filledElementsCounter < blocks) 
    {
        // blocks until the value changes, this moves sequentially 
        // while threads have no order (fine for this sample).
        while(hResult[filledElementsCounter] == 0);
        printf("%f\n", hResult[filledElementsCounter]);;
        filledElementsCounter++;
    }
    cudaFreeHost((void *)hResult);
    system("pause");
}

目前，此示例将无限期地等待，因为除非我发出cudaDeviceSynchronize，否则不会从设备中读取任何内容。下面的示例可以工作，但它是而不是，因为它违背了异步复制的目的：

void main()
{
    size_t blocks = 2;
    volatile float* hResult;
    cudaHostAlloc((void**)&hResult, blocks*sizeof(float), cudaHostAllocMapped);
    Kernel<<<1,blocks>>>(hResult);
    cudaError_t error = cudaDeviceSynchronize();
    if (error != cudaSuccess) { throw; }
    for(int i = 0; i < blocks; i++) 
    {
        printf("%f\n", hResult[i]);
    }
    cudaFreeHost((void *)hResult);
    system("pause");
}

cuda

回答 3

Stack Overflow用户

发布于 2013-10-15 04:27:10

我在CentOS6.2上使用CUDA 5.5和Tesla M2090播放了您的代码，最后可以得出这样的结论：

它不能在您的系统上工作的问题必须是驱动程序问题，我建议您获得TCC驱动程序。

我附加了我的代码，运行良好，做你想做的事。这些值在内核结束之前出现在主机端。如您所见，我添加了一些计算代码，以防止由于编译器优化而删除for循环。我添加了一个流和一个回调，在流中的所有工作完成后执行。该程序输出1 2，在很长一段时间内，在stream finished...被打印到控制台之前什么也不做。

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

 #define SEC_CUDA_CALL(val)           checkCall  ( (val), #val, __FILE__, __LINE__ )

 bool checkCall(cudaError_t result, char const* const func,  const char *const file, int const line)
 {
    if (result != cudaSuccess)
    {
            std::cout << "CUDA (runtime api) error: " << func << " failed! " << cudaGetErrorString(result) << " (" << result << ") " << file << ":" << line << std::endl;
    }
    return result != cudaSuccess;
}

class Callback
{
public:
    static void CUDART_CB dispatch(cudaStream_t stream, cudaError_t status, void *userData);

private:
    void call();
};

void CUDART_CB Callback::dispatch(cudaStream_t stream, cudaError_t status, void *userData)
{
    Callback* cb = (Callback*) userData;
    cb->call();
}

void Callback::call()
{
     std::cout << "stream finished..." << std::endl;
}



__global__ void Kernel(volatile float* hResult)
{
    int tid = threadIdx.x + blockIdx.x * blockDim.x;

    hResult[tid] = tid + 1;
    __threadfence_system();
    float A = 0;
    for (int timeWater = 0; timeWater  < 100000000; timeWater++)
    {
        A = sin(cos(log(hResult[0] * hResult[1]))) + A;
        A = sqrt(A);
    }
}

int main(int argc, char* argv[])
{
    size_t blocks = 2;
    volatile float* hResult;
    SEC_CUDA_CALL(cudaHostAlloc((void**)&hResult,blocks*sizeof(float),cudaHostAllocMapped));

    cudaStream_t stream;
    SEC_CUDA_CALL(cudaStreamCreateWithFlags(&stream, cudaStreamNonBlocking));
    Callback obj;
    Kernel<<<1,blocks,NULL,stream>>>(hResult);
    SEC_CUDA_CALL(cudaStreamAddCallback(stream, Callback::dispatch, &obj, 0));

    int filledElementsCounter = 0;

    while (filledElementsCounter < blocks)
    {
        while(hResult[filledElementsCounter] == 0);
        std::cout << hResult[filledElementsCounter] << std::endl;
        filledElementsCounter++;
    }

    SEC_CUDA_CALL(cudaStreamDestroy(stream));
    SEC_CUDA_CALL(cudaFreeHost((void *)hResult));
}

没有电话返回一个错误和库达-模检查没有发现任何问题。这是按计划进行的。你真该试试TCC的驱动程序。

票数 4

Stack Overflow用户

发布于 2013-05-07 03:16:04

不能将主机指针直接传递给内核。如果使用带有cudaHostAlloc标志的cudaHostAllocMapped分配主机内存，那么首先必须检索映射主机内存的设备指针，然后才能在内核中使用它。使用cudaHostGetDevicePointer获取映射主机内存的设备指针。

float* hResult, *dResult;
cudaHostAlloc((void**)&hResult, blocks*sizeof(float), cudaHostAllocMapped);
cudaHostGetDevicePointer(&dResult,hResult);
Kernel<<<1,blocks>>>(dResult);