aboutsummaryrefslogtreecommitdiff
path: root/mlir/lib/CAPI/Dialect/Complex.cpp
blob: ddf648b8b705cd20567edf0a0cc76bf5d0c28c0c (plain)
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
36
37
38
39
40
41
42
43
44
45
46
//===- Complex.cpp - C Interface for Complex dialect ----------------------===//
//
// 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-c/Dialect/Complex.h"
#include "mlir-c/IR.h"
#include "mlir-c/Support.h"
#include "mlir/CAPI/Registration.h"
#include "mlir/Dialect/Complex/IR/Complex.h"

using namespace mlir;

MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(Complex, complex,
                                      mlir::complex::ComplexDialect)

bool mlirAttributeIsAComplex(MlirAttribute attr) {
  return isa<complex::NumberAttr>(unwrap(attr));
}

MlirAttribute mlirComplexAttrDoubleGet(MlirContext ctx, MlirType type,
                                       double real, double imag) {
  return wrap(
      complex::NumberAttr::get(cast<ComplexType>(unwrap(type)), real, imag));
}

MlirAttribute mlirComplexAttrDoubleGetChecked(MlirLocation loc, MlirType type,
                                              double real, double imag) {
  return wrap(complex::NumberAttr::getChecked(
      unwrap(loc), cast<ComplexType>(unwrap(type)), real, imag));
}

double mlirComplexAttrGetRealDouble(MlirAttribute attr) {
  return cast<complex::NumberAttr>(unwrap(attr)).getReal().convertToDouble();
}

double mlirComplexAttrGetImagDouble(MlirAttribute attr) {
  return cast<complex::NumberAttr>(unwrap(attr)).getImag().convertToDouble();
}

MlirTypeID mlirComplexAttrGetTypeID(void) {
  return wrap(complex::NumberAttr::getTypeID());
}