test_util.cuh

/*
 * 版权所有 (c) 2022-2025, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）授权；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可在以下网址获取许可证副本
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证以“按原样”基础分发，
 * 不附带任何明示或默示的保证或条件。
 * 请参阅许可证以了解管理权限和
 * 限制的具体语言。
 */
 
#pragma once
 
#include <cuspatial/geometry/vec_2d.hpp>
#include <cuspatial/traits.hpp>
 
#include <rmm/device_uvector.hpp>
 
#include <thrust/for_each.h>
#include <thrust/host_vector.h>
 
#include <cstdio>
#include <iomanip>
#include <string_view>
 
namespace cuspatial {
 
namespace test {

template <typename T, typename Vector>
thrust::host_vector<T> to_host(Vector const& dvec)
{
 if constexpr (std::is_same_v<Vector, rmm::device_uvector<T>>) {
 thrust::host_vector<T> hvec(dvec.size());
 cudaMemcpyAsync(hvec.data(),
 dvec.data(),
 dvec.size() * sizeof(T),
 cudaMemcpyKind::cudaMemcpyDeviceToHost,
 dvec.stream());
 dvec.stream().synchronize();
 return hvec;
 } else {
 return thrust::host_vector<T>(dvec);
  }
}

template <typename Iter, typename T = cuspatial::iterator_value_type<Iter>>
thrust::host_vector<T> to_host(Iter begin, Iter end)
{
 return thrust::host_vector<T>(begin, end);
}

template <typename Iter>
void print_device_range(Iter begin,
 Iter end,
 std::string_view pre = "",
 std::string_view post = "\n")
{
 auto hvec = to_host(begin, end);
 
 std::cout << pre;
 std::for_each(hvec.begin(), hvec.end(), [](auto const& x) { std::cout << x << " "; });
 std::cout << post;
}

template <typename Vector>
void print_device_vector(Vector const& vec, std::string_view pre = "", std::string_view post = "\n")
{
 using T = typename Vector::value_type;
 auto hvec = to_host<T>(vec);
 
 std::cout << pre;
 std::for_each(hvec.begin(), hvec.end(), [](auto const& x) { std::cout << x << " "; });
 std::cout << post;
}
 
} // namespace test
} // namespace cuspatial