使用thrust::unique处理float3元组时结果异常的技术问询

看起来你碰到了Thrust中unique函数处理自定义类型的常见坑，我帮你分析下问题根源和解决方案：

核心问题分析 #

Thrust的thrust::unique和标准库的std::unique行为一致，有两个关键前提很容易被忽略：

1. 仅移除连续的重复元素：如果你的数组没有预先排序，非相邻的重复元素不会被去除，这是最常见的原因。
2. 浮点数精确比较的风险：直接用==比较float类型的值，可能因为计算过程中的微小精度误差（比如浮点运算的舍入），导致本该视为相等的float3被判定为不等。

另外，虽然你定义了全局的operator==用于float3，但Thrust在使用自定义谓词时，显式传入谓词会更可靠，避免编译器或Thrust内部的重载解析问题。

修正后的完整代码 #

#include <iostream>
#include <cmath>
#include <thrust/tuple.h>
#include <thrust/device_vector.h>
#include <thrust/unique.h>
#include <thrust/sort.h>

// 自定义float3相等谓词（带epsilon容错）
__host__ __device__ __forceinline__ bool float3_equal(const float3& a, const float3& b, float eps = 1e-6f) {
    return (fabs(a.x - b.x) < eps) && 
           (fabs(a.y - b.y) < eps) && 
           (fabs(a.z - b.z) < eps);
}

// 自定义float3排序谓词（用于前置排序）
__host__ __device__ __forceinline__ bool float3_less(const float3& a, const float3& b) {
    if (a.x != b.x) return a.x < b.x;
    if (a.y != b.y) return a.y < b.y;
    return a.z < b.z;
}

int main() {
    // 示例输入：包含重复和非重复的float3元素
    thrust::host_vector<float3> h_vec = {
        make_float3(1.0f, 2.0f, 3.0f),
        make_float3(4.0f, 5.0f, 6.0f),
        make_float3(1.0f, 2.0f, 3.0f), // 重复元素
        make_float3(7.0f, 8.0f, 9.0f),
        make_float3(4.0f, 5.0f, 6.0f), // 重复元素
        make_float3(1.0f, 2.0f, 3.0000001f) // 近似相等的元素
    };

    // 拷贝到设备端
    thrust::device_vector<float3> d_vec = h_vec;

    // 第一步：先排序，让重复元素连续
    thrust::sort(d_vec.begin(), d_vec.end(), float3_less);

    // 第二步：调用unique，使用自定义的相等谓词
    auto new_end = thrust::unique(d_vec.begin(), d_vec.end(), float3_equal);

    // 移除末尾的无效元素
    d_vec.erase(new_end, d_vec.end());

    // 输出结果
    std::cout << "去重后的元素数量：" << d_vec.size() << std::endl;
    for (const auto& elem : d_vec) {
        std::cout << "(" << elem.x << ", " << elem.y << ", " << elem.z << ")" << std::endl;
    }

    return 0;
}

关键步骤说明 #

• 前置排序：调用thrust::sort将数组排序，确保所有重复元素（或近似相等元素）相邻，这样unique才能正确识别并移除它们。
• 带epsilon的相等判断：用fabs(a.x - b.x) < eps替代直接的==，容忍浮点运算带来的微小精度误差，你可以根据需求调整eps的大小（比如1e-6或1e-9）。
• 显式传入谓词：在sort和unique中显式传入自定义的比较/相等谓词，避免依赖全局运算符的重载解析，让代码更清晰可靠。

额外注意事项 #

• 如果你的float3是完全精确的（比如直接初始化的常量，没有经过浮点计算），可以保留精确的==比较，但仍然需要先排序。
• Thrust的unique返回的是去重后范围的末尾迭代器，需要调用erase来真正缩小容器的大小。

内容的提问来源于stack exchange，提问作者Vitality