resource_types.hpp

 
#pragma once
 
#include <cuda/memory_resource>
 
namespace rapidsmpf {
 
using any_device_resource = cuda::mr::any_resource<cuda::mr::device_accessible>;
 
using any_host_device_resource =
    cuda::mr::any_resource<cuda::mr::host_accessible, cuda::mr::device_accessible>;
 
using any_host_resource = cuda::mr::any_resource<cuda::mr::host_accessible>;
 
template <typename... Properties>
[[nodiscard]] bool is_host_accessible(
    cuda::mr::resource_ref<Properties...> const& mr
) noexcept {
    auto const accessibility =
        get_property(mr, cuda::mr::dynamic_accessibility_property{});
    return accessibility == cuda::mr::__memory_accessibility::__host
           || accessibility == cuda::mr::__memory_accessibility::__host_device;
}
 
template <typename... Properties>
[[nodiscard]] bool is_device_accessible(
    cuda::mr::resource_ref<Properties...> const& mr
) noexcept {
    // See `is_host_accessible` for why the call is unqualified (ADL).
    auto const accessibility =
        get_property(mr, cuda::mr::dynamic_accessibility_property{});
    return accessibility == cuda::mr::__memory_accessibility::__device
           || accessibility == cuda::mr::__memory_accessibility::__host_device;
}
 
}  // namespace rapidsmpf