BufferEventHandler.hpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//
 
#pragma once
 
#include "shambase/SourceLocation.hpp"
#include "shambackends/EventList.hpp"
#include "shambackends/sycl.hpp"
#include <shambackends/details/BufferEventHandler.hpp>
 
namespace sham::details {
 
    class BufferEventHandler {
 
        public:
        std::vector<sycl::event> read_events;
 
        std::vector<sycl::event> write_events;
 
        inline void wait_all(SourceLocation src_loc = SourceLocation{}) {
            sycl::event::wait(read_events);
            sycl::event::wait(write_events);
 
            read_events.clear();
            write_events.clear();
        }
 
        inline bool is_empty() { return read_events.empty() && write_events.empty(); }
 
        enum last_op { READ, WRITE } last_access;
 
        bool up_to_date_events = true;
 
        SourceLocation last_access_loc;
 
        void read_access(sham::EventList &depends_list, SourceLocation src_loc = SourceLocation{});
 
        void write_access(sham::EventList &depends_list, SourceLocation src_loc = SourceLocation{});
 
        void complete_state(sycl::event e, SourceLocation src_loc = SourceLocation{});
 
        void complete_state(
            const std::vector<sycl::event> &events, SourceLocation src_loc = SourceLocation{});
 
        void filter_events();
 
        inline void complete_state(sham::EventList &events) {
            complete_state(events.events);
            events.consumed = true;
        }
    };
 
} // namespace sham::details