DeviceQueue.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/memory.hpp"
#include "shambackends/Device.hpp"
#include "shambackends/DeviceContext.hpp"
#include "shambackends/EventList.hpp"
 
namespace sham {
 
    class DeviceQueue {
        public:
        std::shared_ptr<DeviceContext> ctx;
 
        sycl::queue q;
 
        std::string queue_name;
 
        bool in_order;
 
        bool wait_after_submit = false;
 
        DeviceQueue(std::string queue_name, std::shared_ptr<DeviceContext> ctx, bool in_order);
 
        template<class Fct>
        sycl::event submit(Fct &&fct) {
 
            auto e = q.submit([&](sycl::handler &h) {
                fct(h);
            });
 
            if (wait_after_submit) {
                e.wait_and_throw();
            }
 
            return e;
        }
 
        template<class Fct>
        sycl::event submit(EventList &elist, Fct &&fct) {
 
            elist.consumed = true;
 
            auto e = q.submit([&](sycl::handler &h) {
                elist.apply_dependency(h);
                fct(h);
            });
 
            if (wait_after_submit) {
                e.wait_and_throw();
            }
 
            return e;
        }
 
        inline DeviceProperties &get_device_prop() {
            return shambase::get_check_ref(ctx).device->prop;
        }
    };
 
} // namespace sham