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

Updated to add opbase

Browse source
This commit is contained in:
Alejandro Saucedo 2020-11-01 16:27:59 +00:00
parent 0e9ba00b71
commit 6afe6463c2
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
python/src/main.cpp
View file

@ -21,7 +21,10 @@ PYBIND11_MODULE(komputepy, m) {
.def(py::init(
[](const std::vector<float>& data, kp::Tensor::TensorTypes tensorTypes) {
return std::unique_ptr<kp::Tensor>(new kp::Tensor(data, tensorTypes));
}));
}))
.def("data", &kp::Tensor::data);
py::class_<kp::OpBase>(m, "OpBase");
#ifdef VERSION_INFO
m.attr("__version__") = VERSION_INFO;