nomic-ai/kompute
1
0
Fork 0
mirror of https://github.com/nomic-ai/kompute.git synced 2026-05-11 17:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

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;