mirror of https://github.com/llvm/torch-mlir
29 lines
905 B
C++
29 lines
905 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 "pybind.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
|