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

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");