mlir/lib/CAPI/IR/DialectHandle.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//===- DialectHandle.cpp - C Interface for MLIR Dialect Operations -------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "mlir/CAPI/Registration.h"

static inline const MlirDialectRegistrationHooks *
unwrap(MlirDialectHandle handle) {
  return (const MlirDialectRegistrationHooks *)handle.ptr;
}

MlirStringRef mlirDialectHandleGetNamespace(MlirDialectHandle handle) {
  return unwrap(handle)->getNamespaceHook();
}

void mlirDialectHandleInsertDialect(MlirDialectHandle handle,
                                    MlirDialectRegistry registry) {
  unwrap(handle)->insertHook(registry);
}

void mlirDialectHandleRegisterDialect(MlirDialectHandle handle,
                                      MlirContext ctx) {
  mlir::DialectRegistry registry;
  mlirDialectHandleInsertDialect(handle, wrap(&registry));
  unwrap(ctx)->appendDialectRegistry(registry);
}

MlirDialect mlirDialectHandleLoadDialect(MlirDialectHandle handle,
                                         MlirContext ctx) {
  return unwrap(handle)->loadHook(ctx);
}