base_fixture.hpp

/*
 * 版权所有 (c) 2022-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版本 ("许可证") 进行许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可以获得许可证的副本，网址如下：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或经书面同意，否则软件
 * 根据许可证分发，按“原样”提供，
 * 不附带任何明示或暗示的担保或条件。
 * 请参阅许可证，了解管理权限和
 * 限制条件的具体语言。
 */
 
#pragma once
 
#include <cudf_test/cudf_gtest.hpp>
 
#include <rmm/cuda_stream_view.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
 
namespace cuspatial {
namespace test {

class RMMResourceMixin {
 rmm::device_async_resource_ref _mr{rmm::mr::get_current_device_resource()};
 rmm::cuda_stream_view _stream{rmm::cuda_stream_default};
 
 public
 rmm::device_async_resource_ref mr() { return _mr; }

 rmm::cuda_stream_view stream() { return _stream; }
};

class BaseFixture : public RMMResourceMixin, public ::testing::Test {};

template <typename... Ts>
class BaseFixtureWithParam : public RMMResourceMixin,
 public ::testing::TestWithParam<std::tuple<Ts...>> {};

using FloatingPointTypes = ::testing::Types<float, double>;
 
} // namespace test
} // namespace cuspatial