[PYTHON] Some cleaning of the PyBind11 wrappers (#62)
This commit is contained in:
committed by
Philippe Tillet
parent
08909b49c8
commit
2a02fabdac
12
python/src/main.cc
Normal file
12
python/src/main.cc
Normal 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);
|
||||
}
|
Reference in New Issue
Block a user