mirror of https://github.com/llvm/torch-mlir
25 lines
727 B
C++
25 lines
727 B
C++
//===- 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);
|
|
|
|
} // namespace torch_mlir
|
|
|
|
#endif
|