c++ - 如何将附加数组传递给 Thrust 的 min_element 谓词

Question

我正在尝试使用 Thrust 的 min_element 缩减来找到 Prim 算法中的下一个边缘。我遍历图形边缘。这是我的比较函数:

struct compareEdge {
    __host__ /*__device__*/ bool operator()(Edge l, Edge r) {
        if (visited[l.u] != visited[l.v] && visited[r.u] != visited[r.v]) {
            return l.cost < r.cost;
        } else if (visited[l.u] != visited[l.v]) {
            return true;
        } else {
            return false;
        }
    }
};

不幸的是，这段代码无法在设备上运行，因为我使用了 visited 数组，我在其中标记了已经访问过的节点。我如何将这个数组传递给我的谓词以使其可用于设备执行的代码？

Answer 1

可能有多种方法可以处理此问题。我将介绍一种方法。请注意，您的问题是如何将任意数据集传递给仿函数，这正是我要展示的内容。我不是要解决您提出的仿函数是否是 thrust::min_element 的有用比较谓词的问题。 (我不确定)。

一种方法是简单地拥有一个静态定义的数组:

__device__ int d_visited[DSIZE];

然后在您的主机代码中，在使用仿函数之前，您需要初始化数组:

cudaMemcpyToSymbol(d_visited, visited, DSIZE*sizeof(int));

您的仿函数代码必须修改。由于您可能希望仿函数在主机或设备上可用，因此我们需要基于此控制代码:

struct compareEdge {
    __host__ __device__ bool operator()(Edge l, Edge r) {
#ifdef __CUDA_ARCH__
        if (d_visited[l.u] != d_visited[l.v] && d_visited[r.u] != d_visited[r.v]) {
            return l.cost < r.cost;
        } else if (d_visited[l.u] != d_visited[l.v]) {
            return true;
        } else {
            return false;
        }
#else
        if (visited[l.u] != visited[l.v] && visited[r.u] != visited[r.v]) {
            return l.cost < r.cost;
        } else if (visited[l.u] != visited[l.v]) {
            return true;
        } else {
            return false;
        }
#endif
    }
};