mirror of https://github.com/llvm/torch-mlir
33 lines
1.2 KiB
C
33 lines
1.2 KiB
C
|
/*===-- torch-mlir-c/Registration.h - Registration functions -----*- C -*-===*\
|
||
|
|
|* *|
|
||
|
|
|* Part of the LLVM Project, under the Apache License v2.0 with LLVM *|
|
||
|
|
|* Exceptions. *|
|
||
|
|
|* See https://llvm.org/LICENSE.txt for license information. *|
|
||
|
|
|* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception *|
|
||
|
|
|* *|
|
||
|
|
\*===----------------------------------------------------------------------===*/
|
||
|
|
|
||
|
|
#ifndef TORCHMLIR_C_REGISTRATION_H
|
||
|
|
#define TORCHMLIR_C_REGISTRATION_H
|
||
|
|
|
||
|
|
#include "mlir-c/IR.h"
|
||
|
|
#include "mlir-c/Support.h"
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/** Registers all dialects with a context.
|
||
|
|
* This is needed before creating IR for these Dialects.
|
||
|
|
*/
|
||
|
|
MLIR_CAPI_EXPORTED void torchMlirRegisterAllDialects(MlirContext context);
|
||
|
|
|
||
|
|
/** Registers all passes for symbolic access with the global registry. */
|
||
|
|
MLIR_CAPI_EXPORTED void torchMlirRegisterAllPasses();
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif // TORCHMLIR_C_REGISTRATION_H
|