direct_kernel.hpp

Go to the documentation of this file.

#pragma once

 
#include <cstdint>
#include <span>
#include <string_view>
 
#include "orrery/core/softening.hpp"
#include "orrery/core/types.hpp"
#include "orrery/core/vec3.hpp"
#include "orrery/core/vec3_span.hpp"
 
namespace orrery::solvers {

enum class KernelKind : std::uint8_t {
    kScalar,

    kAvx2,
};

using AccumulateRange = core::Vec3 (*)(core::Vec3Span<const core::Real> positions,
                                       std::span<const core::Real> masses, core::Vec3 target,
                                       core::Index begin, core::Index end,
                                       core::Softening softening) noexcept;

[[nodiscard]] core::Vec3 accumulate_range_scalar(core::Vec3Span<const core::Real> positions,
                                                 std::span<const core::Real> masses,
                                                 core::Vec3 target, core::Index begin,
                                                 core::Index end,
                                                 core::Softening softening) noexcept;
 
#ifdef ORRERY_HAS_AVX2_KERNEL
[[nodiscard]] core::Vec3 accumulate_range_avx2(core::Vec3Span<const core::Real> positions,
                                               std::span<const core::Real> masses,
                                               core::Vec3 target, core::Index begin,
                                               core::Index end, core::Softening softening) noexcept;
#endif

[[nodiscard]] bool kernel_available(KernelKind kind) noexcept;

[[nodiscard]] KernelKind fastest_available_kernel() noexcept;

[[nodiscard]] AccumulateRange accumulate_range_for(KernelKind kind) noexcept;

[[nodiscard]] core::Index kernel_lane_count(KernelKind kind) noexcept;

[[nodiscard]] std::string_view to_string(KernelKind kind) noexcept;
 
} // namespace orrery::solvers