diff --git a/python/src/main.cpp b/python/src/main.cpp index f2bb13090..d0447fe8e 100644 --- a/python/src/main.cpp +++ b/python/src/main.cpp @@ -220,7 +220,7 @@ PYBIND11_MODULE(kp, m) { py::arg("workgroup") = kp::Workgroup(), py::arg("spec_consts") = std::vector(), py::arg("push_consts") = std::vector()) - .def("algorithm_t", [np](kp::Manager& self, + .def("algorithm", [np](kp::Manager& self, const std::vector>& tensors, const py::bytes& spirv, const kp::Workgroup& workgroup, diff --git a/python/test/test_kompute.py b/python/test/test_kompute.py index 385933f26..8660b0151 100644 --- a/python/test/test_kompute.py +++ b/python/test/test_kompute.py @@ -229,7 +229,7 @@ def test_pushconsts_int(): spec_consts = np.array([], dtype=np.int32) push_consts = np.array([-1, -1, -1], dtype=np.int32) - algo = mgr.algorithm_t([tensor], spirv, (1, 1, 1), spec_consts, push_consts) + algo = mgr.algorithm([tensor], spirv, (1, 1, 1), spec_consts, push_consts) (mgr.sequence() .record(kp.OpTensorSyncDevice([tensor]))