1#include <pybind11/pybind11.h>
2
3void init_superblocking(pybind11::module &m);
4void init_torch_utils(pybind11::module &m);
5void init_triton(pybind11::module &m);
6void init_cutlass(pybind11::module &m);
7
8PYBIND11_MODULE(libtriton, m) {
9 m.doc() = "Python bindings to the C++ Triton API";
10 init_triton(m);
11 init_superblocking(m);
12#ifdef WITH_CUTLASS_BINDINGS
13 init_cutlass(m);
14#endif
15}
16