device_span.cuh

Go to the documentation of this file.

#pragma once
 
#include <cuda/std/span>
 
#include <cstddef>
 
namespace cusbf {
 
template <typename T>
struct device_span : cuda::std::span<T> {
    using cuda::std::span<T>::span;
 
    template <typename U>
        requires std::is_convertible_v<U (*)[], T (*)[]>
    constexpr explicit device_span(device_span<U> other) noexcept
        : cuda::std::span<T>(other.data(), other.size()) {}
};
 
}  // namespace cusbf