主机上运行的自定义函数的Thrust结果不正确

Incorrect results from Thrust for custom functor run on host

本文关键字：结果 不正确 Thrust 自定义函数 运行 主机      更新时间：2023-10-16

当我试图实现任何函子时，都会得到糟糕的结果。例如，我尝试了一个类似于thrust::negate<T>的否定函子以下是使用内置否定函子产生良好结果的示例代码：

int data[10] = {-5, 0, 2, -3, 2, 4, 0, -1, 2, 8};
int out[10];
thrust::negate<int> op;
thrust::transform(data, data + 10, out, op);

out变量变为{5, 0, -2, 3, -2, -4, 0, 1, -2, -8}，但当我实现自己的函子（如）时

struct NegateFunctor{
__host__ __device__
    int operator()(const int &val) const {
        return -val;
}
};

并称其为thrust::transform(data, data + 10, out, NegateFunctor()) out包含{-858993460, -858993460, -858993460, -858993460, -858993460, -858993460, -858993460, -858993460, -858993460, -858993460}

我在64位计算机上使用Visual Studio 2010 5.0 CUDA。

感谢

如果我编译并运行您的代码（仅修改为添加打印thrust::transform调用的结果）：

#include <thrust/transform.h>
#include <thrust/functional.h>
struct NegateFunctor{
    __host__ __device__
        int operator()(const int &val) const {
            return -val;
    }
};
int main(int argc, char** argv){
    int data[10] = {-5, 0, 2, -3, 2, 4, 0, -1, 2, 8};
    int out[10];
    //thrust::negate<int> op;
    thrust::transform(data, data + 10, out, NegateFunctor());
      for(int i=0; i<10; i++) {
        std::cout << data[i] << " " << out[i] << std::endl;
    }
    return 0;
}

我得到了这个（操作系统10.6.7上的CUDA 5.0）：

$ nvcc thrust_neg2.cu 
$ ./a.out
-5 5
0 0
2 -2
-3 3
2 -2
4 -4
0 0
-1 1
2 -2
8 -8

这似乎是正确的。如果您没有看到相同的结果，那么这要么是您正在使用的工具链的特定问题，要么是您没有告诉我们的其他事情导致了问题。

编辑：它出现在您在调试模式下使用nvcc构建的注释中，众所周知，这不适用于Thrust。我建议只发布构建代码。如果问题仍然存在，这可能是给Thrust开发人员的错误报告，而不是堆栈溢出问题。

相关文章：