thrust / testing /regression /nvbug_1990211__scan_requires_assignability_from_zero.fixed1.cu
camenduru's picture
thanks to nvidia ❤
8ae5fc5
#include <thrust/scan.h>
#include <thrust/device_ptr.h>
inline __host__ __device__ uint2 operator+(uint2 a, uint2 b)
{
return make_uint2(a.x + b.x, a.y + b.y);
}
int main() {
int num_elements = 32;
uint2 *input = NULL, *output = NULL;
const uint2 zero = make_uint2(0,0);
thrust::exclusive_scan(thrust::device_ptr<uint2>((uint2*)input),
thrust::device_ptr<uint2>((uint2*)input + num_elements),
thrust::device_ptr<uint2>(output), zero, operator+);
return 0;
}