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

native logging for python

Browse source
This commit is contained in:
alexander-g 2021-01-24 12:12:51 +01:00
parent 661a50e62d
commit 059221a7de
5 changed files with 98 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

14
python/src/main.cpp
View file

@ -2,13 +2,25 @@
#include <pybind11/stl.h>
#include <pybind11/numpy.h>
#include <kompute/Kompute.hpp>
#include <kompute/Core.hpp>
#include <kompute/Manager.hpp>
#include <kompute/operations/OpAlgoBase.hpp>
#include <kompute/operations/OpMult.hpp>
#include <kompute/operations/OpTensorSyncLocal.hpp>
#include <kompute/operations/OpTensorSyncDevice.hpp>
#include <kompute/operations/OpTensorCopy.hpp>
#include <kompute/operations/OpAlgoLhsRhsOut.hpp>
#include "docstrings.hpp"
namespace py = pybind11;
//used in Core.hpp
py::object kp_logger;
PYBIND11_MODULE(kp, m) {
py::module_ logging = py::module_::import("logging");
kp_logger = logging.attr("getLogger")("kp");
logging.attr("basicConfig")();
py::module_ np = py::module_::import("numpy");