13 lines
318 B
C++
13 lines
318 B
C++
|
#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);
|
|||
|
|
}
|