-
Notifications
You must be signed in to change notification settings - Fork 34
Expand file tree
/
Copy pathorder_keeper.cpp
More file actions
43 lines (40 loc) · 1.6 KB
/
order_keeper.cpp
File metadata and controls
43 lines (40 loc) · 1.6 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#include "dpctl4pybind11.hpp"
#include <cstddef>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include "sequential_order_keeper.hpp"
#include <sycl/sycl.hpp>
PYBIND11_MODULE(_seq_order_keeper, m)
{
py::class_<SequentialOrder>(m, "_OrderManager")
.def(py::init<std::size_t>())
.def(py::init<>())
.def(py::init<SequentialOrder>())
.def("get_num_submitted_events",
&SequentialOrder::get_num_submitted_events)
.def("get_num_host_task_events",
&SequentialOrder::get_num_host_task_events)
.def("get_submitted_events", &SequentialOrder::get_submitted_events)
.def("get_host_task_events", &SequentialOrder::get_host_task_events)
.def("add_to_both_events", &SequentialOrder::add_to_both_events)
.def("add_vector_to_both_events",
&SequentialOrder::add_vector_to_both_events)
.def("add_to_host_task_events",
&SequentialOrder::add_to_host_task_events)
.def("add_to_submitted_events",
&SequentialOrder::add_to_submitted_events)
.def("wait", &SequentialOrder::wait,
py::call_guard<py::gil_scoped_release>());
auto submit_empty_task_fn =
[](sycl::queue &exec_q,
const std::vector<sycl::event> &depends) -> sycl::event {
return exec_q.submit([&](sycl::handler &cgh) {
cgh.depends_on(depends);
cgh.single_task([]() {
// empty body
});
});
};
m.def("_submit_empty_task", submit_empty_task_fn, py::arg("sycl_queue"),
py::arg("depends") = py::list());
}