1 | #include <pybind11/pybind11.h> |
---|---|
2 | |
3 | void init_superblocking(pybind11::module &m); |
4 | void init_torch_utils(pybind11::module &m); |
5 | void init_triton(pybind11::module &m); |
6 | void init_cutlass(pybind11::module &m); |
7 | |
8 | PYBIND11_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 |