如何使用 thrust::remove_if 一起检查和删除块 2×i 和 2×i+1

Question

我使用的是 Cuda C++，我有一个大数组Arr ，其中包含如下形式的 64 位无符号整数：

Arr = {a1, b1, a2, b2, ..., an, bn}

Arr中的项目数是2n ，这是一个偶数。 现在，给定一个 boolean function f(int a, int b)，我想知道我是否可以使用 thrust::remove_if 来检查 f(a1,b1), f(a2, b2), ..., f(an, bn ) 并在需要时一起删除两个连续的数字 (ai, bi)？

Answer 1

除了zip_iterator ，我认为更简单的方法是将 64 位整数数组重新解释为thrust::pair数组。 这是一个例子：

$ cat t2157.cu
#include <iostream>
#include <thrust/device_vector.h>
#include <thrust/remove.h>
#include <thrust/sequence.h>
#include <thrust/copy.h>

struct my_remove
{
  template <typename T>
  __host__ __device__
  bool operator()(T t){
    return (thrust::get<0>(t) > thrust::get<1>(t));
  }
};

const size_t n = 32768;
const size_t s = 2*n;
using mt=unsigned long long;
using dt=thrust::pair<mt,mt>;

int main(){

  thrust::device_vector<mt> A(s);
  thrust::sequence(A.begin(), A.end());
  A[0] = 2;  // expecting removal of the first pair
  thrust::copy_n(A.begin(), 6, std::ostream_iterator<mt>(std::cout, ","));
  std::cout << std::endl;
  auto D = thrust::device_pointer_cast<dt>(reinterpret_cast<dt *>(thrust::raw_pointer_cast(A.data())));
  thrust::remove_if(D, D+n, my_remove());
  thrust::copy_n(A.begin(), 6, std::ostream_iterator<mt>(std::cout, ","));
  std::cout << std::endl;
}
$ nvcc -o t2157 t2157.cu
$ compute-sanitizer ./t2157
========= COMPUTE-SANITIZER
2,1,2,3,4,5,
2,3,4,5,6,7,
========= ERROR SUMMARY: 0 errors
$