sycl_direct_solver.hpp

Go to the documentation of this file.

#pragma once

 
#include <memory>
#include <span>
#include <string_view>
 
#include "orrery/backend/sycl_device.hpp"
#include "orrery/backend/worker_statistics.hpp"
#include "orrery/core/softening.hpp"
#include "orrery/core/types.hpp"
#include "orrery/core/vec3_span.hpp"
#include "orrery/solvers/force_solver.hpp"
#include "orrery/solvers/interaction_count.hpp"
 
#ifdef ORRERY_ENABLE_SYCL
 
namespace orrery::solvers {

struct SyclEvaluationTimings {
    backend::Duration staging_in{};

    backend::Duration kernel{};

    backend::Duration staging_out{};
};

class SyclDirectSolver final : public ForceSolver {
public:
    [[nodiscard]] static std::unique_ptr<SyclDirectSolver>
    try_create(core::Softening softening = {});
 
    ~SyclDirectSolver() override;
 
    SyclDirectSolver(const SyclDirectSolver&) = delete;
    SyclDirectSolver& operator=(const SyclDirectSolver&) = delete;
    SyclDirectSolver(SyclDirectSolver&&) noexcept;
    SyclDirectSolver& operator=(SyclDirectSolver&&) noexcept;

    void evaluate(core::Vec3Span<const core::Real> positions, std::span<const core::Real> masses,
                  core::Vec3Span<core::Real> accelerations) override;
 
    [[nodiscard]] std::string_view name() const noexcept override { return "sycl-direct"; }
 
    [[nodiscard]] core::Softening softening() const noexcept override;
 
    [[nodiscard]] InteractionCount interaction_count() const noexcept override;
 
    void reset_interaction_count() noexcept override;

    [[nodiscard]] const backend::DeviceDescription& device() const noexcept;

    [[nodiscard]] core::Index tile_size() const noexcept;

    [[nodiscard]] const SyclEvaluationTimings& timings() const noexcept;

    [[nodiscard]] bool uses_shared_memory() const noexcept;
 
private:
    struct Impl;
 
    explicit SyclDirectSolver(std::unique_ptr<Impl> impl) noexcept;
 
    std::unique_ptr<Impl> impl_;
};
 
} // namespace orrery::solvers
 
#endif // ORRERY_ENABLE_SYCL