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

Added python functionality for shader compile

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-20 19:34:49 +00:00
parent b9ca70b8b1
commit f9c3feccfe
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
python/src/main.cpp
View file

@ -40,8 +40,7 @@ PYBIND11_MODULE(kp, m) {
return py::bytes((const char*)spirv.data(), spirv.size() * sizeof(uint32_t)); return py::bytes((const char*)spirv.data(), spirv.size() * sizeof(uint32_t));
}, },
"Compiles string source provided and returns the value in bytes", "Compiles string source provided and returns the value in bytes",
py::arg("source"), py::arg("entryPoint") = "main", py::arg("definitions") = std::vector<std::pair<std::string,std::string>>() ) py::arg("source"), py::arg("entryPoint") = "main", py::arg("definitions") = std::vector<std::pair<std::string,std::string>>() );
.def_static("compile_sources", &kp::Shader::compile_sources);
py::class_<kp::Tensor, std::shared_ptr<kp::Tensor>>(m, "Tensor", DOC(kp, Tensor)) py::class_<kp::Tensor, std::shared_ptr<kp::Tensor>>(m, "Tensor", DOC(kp, Tensor))
.def(py::init( .def(py::init(