[PYTHON] Some cleaning of the PyBind11 wrappers (#62)

This commit is contained in:
Philippe Tillet
2021-02-06 17:10:44 -08:00
committed by Philippe Tillet
parent 08909b49c8
commit 2a02fabdac
8 changed files with 282 additions and 350 deletions

12
python/src/main.cc Normal file
View File

@@ -0,0 +1,12 @@
#include <pybind11/pybind11.h>
void init_superblocking(pybind11::module &m);
void init_torch_utils(pybind11::module &m);
void init_triton(pybind11::module &m);
PYBIND11_MODULE(libtriton, m) {
m.doc() = "Python bindings to the C++ Triton API";
init_triton(m);
init_torch_utils(m);
init_superblocking(m);
}