MultiRef.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 "shambackends/kernel_call/buffer_access_utils.hpp"
 
namespace sham {
    template<class... Targ>
    struct MultiRef {
        using storage_t = std::tuple<Targ &...>;

        storage_t storage;

        MultiRef(Targ &...arg) : storage(arg...) {}

        auto get_read_access(sham::EventList &depends_list) {
            __shamrock_stack_entry();
            return std::apply(
                [&](auto &...__a) {
                    return std::tuple(details::get_read_access(__a, depends_list)...);
                },
                storage);
        }

        auto get_write_access(sham::EventList &depends_list) {
            __shamrock_stack_entry();
            return std::apply(
                [&](auto &...__a) {
                    return std::tuple(details::get_write_access(__a, depends_list)...);
                },
                storage);
        }

        void complete_event_state(sycl::event e) {
            __shamrock_stack_entry();
            std::apply(
                [&](auto &...__in) {
                    ((details::complete_event_state(__in, e)), ...);
                },
                storage);
        }
    };
} // namespace sham