Function NB_MODULE

python/SiteInitializer.cpp:16–28

using namespace mlir::python::nanobind_adaptors;

NB_MODULE(_site_initialize_1, m) {
  m.doc() = "All CUDA Tile IR related dialects (cuda_tile) and passes.";

  // NB: This is a special API hook that will be automatically called during
  // library initialization.
  m.def("register_dialects", [](MlirDialectRegistry registry) {
    mlirCudaTileRegisterAllDialects(registry);
  });

  // NB: This is not a special API hook and must be invoked manually by a user
  // in Python to register the passes.
  m.def("register_passes", []() { mlirCudaTileRegisterAllPasses(); });
}