thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Updated algo python

Browse source 
Signed-off-by: Alejandro Saucedo <axsauze@gmail.com>
This commit is contained in:
Alejandro Saucedo 2021-09-12 16:56:44 +01:00
parent 932620091c
commit c758ec0c20
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python/src/main.cpp
View file

@ -220,7 +220,7 @@ PYBIND11_MODULE(kp, m) {
py::arg("workgroup") = kp::Workgroup(),
py::arg("spec_consts") = std::vector<float>(),
py::arg("push_consts") = std::vector<float>())
.def("algorithm_t", [np](kp::Manager& self,
.def("algorithm", [np](kp::Manager& self,
const std::vector<std::shared_ptr<kp::Tensor>>& tensors,
const py::bytes& spirv,
const kp::Workgroup& workgroup,

2
python/test/test_kompute.py
View file

@ -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]))