EBGeometry_GPUTransform.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Robert Marskar <robert.marskar@sintef.no>
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef EBGEOMETRY_GPUTRANSFORM_HPP
#define EBGEOMETRY_GPUTRANSFORM_HPP
 
// Std includes
#include <cstddef>
#include <cstdint>
#include <type_traits>
#include <vector>
 
// Our includes
#include "EBGeometry_GPU.hpp"
#include "EBGeometry_GPURuntime.hpp"
#include "EBGeometry_MemoryResource.hpp"
#include "EBGeometry_PODVector.hpp"
 
namespace EBGeometry {
namespace GPU {
 
#if defined(EBGEOMETRY_CUDA) || defined(EBGEOMETRY_HIP)
inline constexpr bool enabled = true;
#else
inline constexpr bool enabled = false;
#endif
 
#if defined(EBGEOMETRY_HIP)
inline constexpr const char* backendName = "HIP";
#elif defined(EBGEOMETRY_CUDA)
inline constexpr const char* backendName = "CUDA";
#else
inline constexpr const char* backendName = "host";
#endif
 
inline constexpr unsigned int DefaultBlockSize = 128;
 
inline constexpr unsigned int DefaultMinResidentThreads = 384;
 
[[nodiscard]] EBGEOMETRY_HOST
inline bool
deviceAvailable() noexcept;
 
[[nodiscard]] EBGEOMETRY_HOST
inline MemoryResource&
memoryResource() noexcept;
 
template <class Query, class In>
using TransformResult = std::decay_t<std::invoke_result_t<const Query&, const In&>>;
 
template <unsigned int BlockSize          = DefaultBlockSize,
          unsigned int MinResidentThreads = DefaultMinResidentThreads,
          class F>
EBGEOMETRY_HOST
inline void
parallelFor(std::size_t a_n, const F& a_f);
 
template <unsigned int BlockSize          = DefaultBlockSize,
          unsigned int MinResidentThreads = DefaultMinResidentThreads,
          class Query,
          class In,
          class Out>
EBGEOMETRY_HOST
inline void
transform(const Query& a_query, PODSpan<In> a_in, PODSpan<Out> a_out);
 
template <unsigned int BlockSize          = DefaultBlockSize,
          unsigned int MinResidentThreads = DefaultMinResidentThreads,
          class Query,
          class In,
          class Out>
EBGEOMETRY_HOST
inline void
transform(const Query& a_query, const Pool& a_pool, const PODVector<In>& a_in, const PODVector<Out>& a_out);
 
template <unsigned int BlockSize          = DefaultBlockSize,
          unsigned int MinResidentThreads = DefaultMinResidentThreads,
          class Query,
          class In,
          class Out>
EBGEOMETRY_HOST
inline void
transform(const Query& a_query, const std::vector<In>& a_in, std::vector<Out>& a_out);
 
} // namespace GPU
} // namespace EBGeometry
 
#include "EBGeometry_GPUTransformImplem.hpp"
 
#endif