From fb8acb5b3feae139afd725c85287f36dc6c1fe9c Mon Sep 17 00:00:00 2001 From: Olivier BICHLER <olivier.bichler@cea.fr> Date: Mon, 31 Mar 2025 15:47:01 +0200 Subject: [PATCH] Added missing binding --- python_binding/scheduler/pybind_Scheduler.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/python_binding/scheduler/pybind_Scheduler.cpp b/python_binding/scheduler/pybind_Scheduler.cpp index c2fb71103..050af45fd 100644 --- a/python_binding/scheduler/pybind_Scheduler.cpp +++ b/python_binding/scheduler/pybind_Scheduler.cpp @@ -59,6 +59,7 @@ void init_Scheduler(py::module& m){ py::class_<SequentialScheduler, std::shared_ptr<SequentialScheduler>, Scheduler>(m, "SequentialScheduler") .def(py::init<std::shared_ptr<GraphView>&, bool>(), py::arg("graph_view"), py::arg("reset_cp_model") = true) + .def("set_scheduling_policy", &SequentialScheduler::setSchedulingPolicy, py::arg("policy")) .def("get_sequential_static_scheduling", &SequentialScheduler::getSequentialStaticScheduling, py::arg("step") = 0) .def("generate_memory", &SequentialScheduler::generateMemory, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false) .def("generate_memory_auto_concat", &SequentialScheduler::generateMemoryAutoConcat, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false) -- GitLab