| 52 | } |
| 53 | |
| 54 | static void BindConfig(py::module &m) { |
| 55 | py::enum_<core::BlasProvider>(m, "BlasProvider") |
| 56 | .value("MKL", core::BlasProvider::MKL) |
| 57 | .value("OpenBlas", core::BlasProvider::OpenBlas); |
| 58 | |
| 59 | m.def("is_compiled_with_cuda", &core::IsCompiledWithCUDA) |
| 60 | .def("get_blas_provider", &core::GetBlasProvider); |
| 61 | } |
| 62 | |
| 63 | PYBIND11_MODULE(turbo_transformers_cxx, m) { |
| 64 | char *argv[] = {strdup("turbo_transformers_cxx"), nullptr}; |