//===- init_python_bindings.h -----------------------------------*- C++ -*-===//
//
// This file is licensed under a pytorch-style license
// See frontends/pytorch/LICENSE for license information.
//
//===----------------------------------------------------------------------===//

#ifndef INIT_PYTHON_BINDINGS_H
#define INIT_PYTHON_BINDINGS_H

#include <pybind11/pybind11.h>

namespace torch_mlir {

// Perform top-level initialization for the module.
void InitBindings(pybind11::module &m);

// Adds bindings related to the type-dispatch program capture mechanism.
// Only defined if NPCOMP_ENABLE_TORCH_TYPE_DISPATCH (optional feature).
void InitTypeDispatchBindings(pybind11::module &m);

// Adds bindings related to the c10-dispatch program capture mechanism.
// Only defined if !NPCOMP_ENABLE_TORCH_TYPE_DISPATCH (default).
void InitC10DispatchBindings(pybind11::module &m);

} // namespace torch_mlir

#endif