nomic-ai/kompute
1
0
Fork 0
mirror of https://github.com/nomic-ai/kompute.git synced 2026-05-12 01:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

caching logging functions + single include rebuild

Browse Source
This commit is contained in:
alexander-g
2021-01-24 15:44:07 +01:00
parent 059221a7de
commit a0f9af965a
3 changed files with 2101 additions and 2082 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
python/src/main.cpp
View File

@@ -2,24 +2,21 @@
#include <pybind11/stl.h>
#include <pybind11/numpy.h>
#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 <AggregateHeaders.cpp>
#include "docstrings.hpp"
namespace py = pybind11;
//used in Core.hpp
py::object kp_logger;
py::object kp_debug, kp_info, kp_warning, kp_error;
PYBIND11_MODULE(kp, m) {
py::module_ logging = py::module_::import("logging");
kp_logger = logging.attr("getLogger")("kp");
py::module_ logging = py::module_::import("logging");
py::object kp_logger = logging.attr("getLogger")("kp");
kp_debug = kp_logger.attr("debug");
kp_info = kp_logger.attr("info");
kp_warning = kp_logger.attr("warning");
kp_error = kp_logger.attr("error");
logging.attr("basicConfig")();
py::module_ np = py::module_::import("numpy");