推力仿函数:"要求推出太多资源"

时间:2012-04-11 08:27:57

标签: cuda thrust

我正试图在CUDA中实现类似的东西:

每个元素

p = { p if p >= floor
      z if p < floor

floorz是在测试开始时配置的常量。

我试图像这样实现它,但是我收到错误“要求启动的资源太多了”

仿函数:

struct floor_functor : thrust::unary_function <float, float>
{
        const float floorLevel, floorVal;

        floor_functor(float _floorLevel, float _floorVal) : floorLevel(_floorLevel), floorVal(_floorVal){}

        __host__
        __device__
        float operator()(float& x) const
        {
            if (x >= floorLevel)
                return x;
            else
                return floorVal;
        }
};

由变换使用:

thrust::transform(input->begin(), input->end(), output.begin(), floor_functor(floorLevel, floorVal));

如果我删除了我的仿函数的一个成员,说floorVal,并使用只有一个成员变量的仿函数,它就可以正常工作。

有谁知道为什么会这样,以及我如何解决它?

其他信息:

我的数组长度为786432个元素。

我的GPU是GeForce GTX590

我正在使用命令构建:

`nvcc -c -g -arch sm_11 -Xcompiler -fPIC -Xcompiler -Wall -DTHRUST_DEBUG  -I <my_include_dir> -o <my_output> <my_source>`

我的cuda版本是4.0:

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2011 NVIDIA Corporation
Built on Thu_May_12_11:09:45_PDT_2011
Cuda compilation tools, release 4.0, V0.2.1221

我的每个块的最大线程数是1024(由deviceQuery报告):

Total amount of constant memory:               65536 bytes
  Total amount of shared memory per block:       49152 bytes
  Total number of registers available per block: 32768
  Warp size:                                     32
  Maximum number of threads per block:           1024
  Maximum sizes of each dimension of a block:    1024 x 1024 x 64
  Maximum sizes of each dimension of a grid:     65535 x 65535 x 65535

UPDATE ::

我偶然发现了我的问题,但是不明白。如果我将我的仿函数从“floor_functor”重命名为其他任何东西,它就可以了!我有没有想法为什么会这样,并且有兴趣听到任何人对此的看法。

1 个答案:

答案 0 :(得分:1)

为了更简单的CUDA实现,您可以在一行代码中使用ArrayFire执行此操作:

p(p < floor) = z;

只需将变量声明为af :: array's。

祝你好运!

免责声明:我从事各种CUDA项目,包括ArrayFire。