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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
//===- translation.c - Test MLIR Target translations ----------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
// RUN: mlir-capi-translation-test 2>&1 | FileCheck %s
#include "llvm-c/Core.h"
#include "llvm-c/Support.h"
#include "llvm-c/Types.h"
#include "mlir-c/BuiltinTypes.h"
#include "mlir-c/Dialect/LLVM.h"
#include "mlir-c/IR.h"
#include "mlir-c/RegisterEverything.h"
#include "mlir-c/Support.h"
#include "mlir-c/Target/LLVMIR.h"
#include <assert.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
// CHECK-LABEL: testToLLVMIR()
static void testToLLVMIR(MlirContext ctx) {
fprintf(stderr, "testToLLVMIR()\n");
LLVMContextRef llvmCtx = LLVMContextCreate();
const char *moduleString = "llvm.func @add(%arg0: i64, %arg1: i64) -> i64 { \
%0 = llvm.add %arg0, %arg1 : i64 \
llvm.return %0 : i64 \
}";
mlirRegisterAllLLVMTranslations(ctx);
MlirModule module =
mlirModuleCreateParse(ctx, mlirStringRefCreateFromCString(moduleString));
MlirOperation operation = mlirModuleGetOperation(module);
LLVMModuleRef llvmModule = mlirTranslateModuleToLLVMIR(operation, llvmCtx);
// clang-format off
// CHECK: define i64 @add(i64 %[[arg1:.*]], i64 %[[arg2:.*]]) {
// CHECK-NEXT: %[[arg3:.*]] = add i64 %[[arg1]], %[[arg2]]
// CHECK-NEXT: ret i64 %[[arg3]]
// CHECK-NEXT: }
// clang-format on
LLVMDumpModule(llvmModule);
LLVMDisposeModule(llvmModule);
mlirModuleDestroy(module);
LLVMContextDispose(llvmCtx);
}
int main(void) {
MlirContext ctx = mlirContextCreate();
mlirDialectHandleRegisterDialect(mlirGetDialectHandle__llvm__(), ctx);
mlirContextGetOrLoadDialect(ctx, mlirStringRefCreateFromCString("llvm"));
testToLLVMIR(ctx);
mlirContextDestroy(ctx);
return 0;
}
|
