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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
//===- transform.c - Test of Transform dialect C API ----------------------===//
//
// 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-transform-test 2>&1 | FileCheck %s
#include "mlir-c/Dialect/Transform.h"
#include "mlir-c/IR.h"
#include "mlir-c/Support.h"
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
// CHECK-LABEL: testAnyOpType
void testAnyOpType(MlirContext ctx) {
fprintf(stderr, "testAnyOpType\n");
MlirType parsedType = mlirTypeParseGet(
ctx, mlirStringRefCreateFromCString("!transform.any_op"));
MlirType constructedType = mlirTransformAnyOpTypeGet(ctx);
assert(!mlirTypeIsNull(parsedType) && "couldn't parse AnyOpType");
assert(!mlirTypeIsNull(constructedType) && "couldn't construct AnyOpType");
// CHECK: equal: 1
fprintf(stderr, "equal: %d\n", mlirTypeEqual(parsedType, constructedType));
// CHECK: parsedType isa AnyOpType: 1
fprintf(stderr, "parsedType isa AnyOpType: %d\n",
mlirTypeIsATransformAnyOpType(parsedType));
// CHECK: parsedType isa OperationType: 0
fprintf(stderr, "parsedType isa OperationType: %d\n",
mlirTypeIsATransformOperationType(parsedType));
// CHECK: !transform.any_op
mlirTypeDump(constructedType);
fprintf(stderr, "\n\n");
}
// CHECK-LABEL: testOperationType
void testOperationType(MlirContext ctx) {
fprintf(stderr, "testOperationType\n");
MlirType parsedType = mlirTypeParseGet(
ctx, mlirStringRefCreateFromCString("!transform.op<\"foo.bar\">"));
MlirType constructedType = mlirTransformOperationTypeGet(
ctx, mlirStringRefCreateFromCString("foo.bar"));
assert(!mlirTypeIsNull(parsedType) && "couldn't parse AnyOpType");
assert(!mlirTypeIsNull(constructedType) && "couldn't construct AnyOpType");
// CHECK: equal: 1
fprintf(stderr, "equal: %d\n", mlirTypeEqual(parsedType, constructedType));
// CHECK: parsedType isa AnyOpType: 0
fprintf(stderr, "parsedType isa AnyOpType: %d\n",
mlirTypeIsATransformAnyOpType(parsedType));
// CHECK: parsedType isa OperationType: 1
fprintf(stderr, "parsedType isa OperationType: %d\n",
mlirTypeIsATransformOperationType(parsedType));
// CHECK: operation name equal: 1
MlirStringRef operationName =
mlirTransformOperationTypeGetOperationName(constructedType);
fprintf(stderr, "operation name equal: %d\n",
mlirStringRefEqual(operationName,
mlirStringRefCreateFromCString("foo.bar")));
// CHECK: !transform.op<"foo.bar">
mlirTypeDump(constructedType);
fprintf(stderr, "\n\n");
}
int main(void) {
MlirContext ctx = mlirContextCreate();
mlirDialectHandleRegisterDialect(mlirGetDialectHandle__transform__(), ctx);
testAnyOpType(ctx);
testOperationType(ctx);
mlirContextDestroy(ctx);
return EXIT_SUCCESS;
}
|
