Skip to content

Commit 346b922

Browse files
author
Michael Gilbert
committed
[bindings] Up-to-date with LayoutLoop update
1 parent 56224de commit 346b922

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

bindings/model.cpp

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,10 +70,22 @@ void BindEngine(py::module& m) {
7070
"Construct and spec Engine.")
7171
.def("spec", &model::Engine::Spec)
7272
.def("pre_evaluation_check", &model::Engine::PreEvaluationCheck)
73-
.def("evaluate", &model::Engine::Evaluate, py::arg("mapping"),
74-
py::arg("workload"), py::arg("sparse_optimizations"),
75-
py::arg("break_on_failure") = true,
76-
py::keep_alive<1, 2>(), py::keep_alive<1, 3>(), py::keep_alive<1, 4>())
73+
.def(
74+
"evaluate",
75+
py::overload_cast<
76+
Mapping&,
77+
problem::Workload&,
78+
sparse::SparseOptimizationInfo*,
79+
bool
80+
>(&model::Engine::Evaluate),
81+
py::arg("mapping"),
82+
py::arg("workload"),
83+
py::arg("sparse_optimizations"),
84+
py::arg("break_on_failure") = true,
85+
py::keep_alive<1, 2>(),
86+
py::keep_alive<1, 3>(),
87+
py::keep_alive<1, 4>()
88+
)
7789
.def("is_evaluated", &model::Engine::IsEvaluated)
7890
.def("utilization", &model::Engine::Utilization)
7991
.def("energy", &model::Engine::Energy)

0 commit comments

Comments
 (0)