xref: /llvm-project-15.0.7/mlir/docs/CAPI.md (revision 4da8fa45)
1# MLIR C API
2
3**Current status: Under development, API unstable, built by default.**
4
5## Design
6
7Many languages can interoperate with C but have a harder time with C++ due to
8name mangling and memory model differences. Although the C API for MLIR can be
9used directly from C, it is primarily intended to be wrapped in higher-level
10language- or library-specific constructs. Therefore the API tends towards
11simplicity and feature minimalism.
12
13**Note:** while the C API is expected to be more stable than C++ API, it
14currently offers no stability guarantees.
15
16### Scope
17
18The API is provided for core IR components (attributes, blocks, operations,
19regions, types, values), Passes and some fundamental type and attribute kinds.
20The core IR API is intentionally low-level, e.g. exposes a plain list of
21operation's operands and attributes without attempting to assign "semantic"
22names to them. Users of specific dialects are expected to wrap the core API in a
23dialect-specific way, for example, by implementing an ODS backend.
24
25### Object Model
26
27Core IR components are exposed as opaque _handles_ to an IR object existing in
28C++. They are not intended to be inspected by the API users (and, in many cases,
29cannot be meaningfully inspected). Instead the users are expected to pass
30handles to the appropriate manipulation functions.
31
32The handle _may or may not_ own the underlying object.
33
34### Naming Convention and Ownership Model
35
36All objects are prefixed with `Mlir`. They are typedefs and should be used
37without `struct`.
38
39All functions are prefixed with `mlir`.
40
41Functions primarily operating on an instance of `MlirX` are prefixed with
42`mlirX`. They take the instance being acted upon as their first argument (except
43for creation functions). For example, `mlirOperationGetNumOperands` inspects an
44`MlirOperation`, which it takes as its first operand.
45
46The *ownership* model is encoded in the naming convention as follows.
47
48-   By default, the ownership is not transferred.
49-   Functions that transfer the ownership of the result to the caller can be in
50    one of two forms:
51    *   functions that create a new object have the name `mlirXCreate<...>`, for
52        example, `mlirOperationCreate`;
53    *   functions that detach an object from a parent object have the name
54        `mlirYTake<...>`, for example `mlirOperationStateTakeRegion`.
55-   Functions that take ownership of some of their arguments have the form
56    `mlirY<...>OwnedX<...>` where `X` can refer to the type or any other
57    sufficiently unique description of the argument, the ownership of which will
58    be taken by the callee, for example `mlirRegionAppendOwnedBlock`.
59-   Functions that create an object by default do not transfer its ownership to
60    the caller, i.e. one of other objects passed in as an argument retains the
61    ownership, they have the form `mlirX<...>Get`. For example,
62    `mlirTypeParseGet`.
63-   Functions that destroy an object owned by the caller are of the form
64    `mlirXDestroy`.
65
66If the code owns an object, it is responsible for destroying the object when it
67is no longer necessary. If an object that owns other objects is destroyed, any
68handles to those objects become invalid. Note that types and attributes are
69owned by the `MlirContext` in which they were created.
70
71### Nullity
72
73A handle may refer to a _null_ object. It is the responsibility of the caller to
74check if an object is null by using `mlirXIsNull(MlirX)`. API functions do _not_
75expect null objects as arguments unless explicitly stated otherwise. API
76functions _may_ return null objects.
77
78### Type Hierarchies
79
80MLIR objects can form type hierarchies in C++. For example, all IR classes
81representing types are derived from `mlir::Type`, some of them may also be also
82derived from common base classes such as `mlir::ShapedType` or dialect-specific
83base classes. Type hierarchies are exposed to C API through naming conventions
84as follows.
85
86-   Only the top-level class of each hierarchy is exposed, e.g. `MlirType` is
87    defined as a type but `MlirShapedType` is not. This avoids the need for
88    explicit upcasting when passing an object of a derived type to a function
89    that expects a base type (this happens more often in core/standard APIs,
90    while downcasting usually involves further checks anyway).
91-   A type `Y` that derives from `X` provides a function `int mlirXIsAY(MlirX)`
92    that returns a non-zero value if the given dynamic instance of `X` is also
93    an instance of `Y`. For example, `int MlirTypeIsAInteger(MlirType)`.
94-   A function that expects a derived type as its first argument takes the base
95    type instead and documents the expectation by using `Y` in its name
96    `MlirY<...>(MlirX, ...)`. This function asserts that the dynamic instance of
97    its first argument is `Y`, and it is the responsibility of the caller to
98    ensure it is indeed the case.
99
100### Returning String References
101
102Numerous MLIR functions return instances of `StringRef` to refer to a non-owning
103segment of a string. This segment may or may not be null-terminated. In C API,
104these functions take an additional callback argument of type
105`MlirStringCallback` (pointer to a function with signature `void (*)(const char
106*, intptr_t, void *)`) and a pointer to user-defined data. This callback is
107invoked with a pointer to the string segment, its size and is forwarded the
108user-defined data. The caller is in charge of managing the string segment
109according to its memory model: for strings owned by the object (e.g., string
110attributes), the caller can store the pointer and the size and use them directly
111as long as the parent object is live or copy the string to a new location with a
112null terminator if expected; for generated strings (e.g., in printing), the
113caller is expected to copy the string segment if it intends to use it later.
114
115**Note:** this interface may be revised in the near future.
116
117### Conversion To String and Printing
118
119IR objects can be converted to a string representation, for example for
120printing, using `mlirXPrint(MlirX, MlirStringCallback, void *)` functions. These
121functions accept take arguments a callback with signature `void (*)(const char
122*, intptr_t, void *)` and a pointer to user-defined data. They call the callback
123and supply it with chunks of the string representation, provided as a pointer to
124the first character and a length, and forward the user-defined data unmodified.
125It is up to the caller to allocate memory if the string representation must be
126stored and perform the copy. There is no guarantee that the pointer supplied to
127the callback points to a null-terminated string, the size argument should be
128used to find the end of the string. The callback may be called multiple times
129with consecutive chunks of the string representation (the printing itself is
130buffered).
131
132*Rationale*: this approach allows the caller to have full control of the
133allocation and avoid unnecessary allocation and copying inside the printer.
134
135For convenience, `mlirXDump(MlirX)` functions are provided to print the given
136object to the standard error stream.
137
138## Common Patterns
139
140The API adopts the following patterns for recurrent functionality in MLIR.
141
142### Indexed Components
143
144An object has an _indexed component_ if it has fields accessible using a
145zero-based contiguous integer index, typically arrays. For example, an
146`MlirBlock` has its arguments as an indexed component. An object may have
147several such components. For example, an `MlirOperation` has attributes,
148operands, regions, results and successors.
149
150For indexed components, the following pair of functions is provided.
151
152-   `intptr_t mlirXGetNum<Y>s(MlirX)` returns the upper bound on the index.
153-   `MlirY mlirXGet<Y>(MlirX, intptr_t pos)` returns 'pos'-th subobject.
154
155The sizes are accepted and returned as signed pointer-sized integers, i.e.
156`intptr_t`. This typedef is available in C99.
157
158Note that the name of subobject in the function does not necessarily match the
159type of the subobject. For example, `mlirOperationGetOperand` returns an
160`MlirValue`.
161
162### Iterable Components
163
164An object has an _iterable component_ if it has iterators accessing its fields
165in some order other than integer indexing, typically linked lists. For example,
166an `MlirBlock` has an iterable list of operations it contains. An object may
167have several iterable components.
168
169For iterable components, the following triple of functions is provided.
170
171-   `MlirY mlirXGetFirst<Y>(MlirX)` returns the first subobject in the list.
172-   `MlirY mlirYGetNextIn<X>(MlirY)` returns the next subobject in the list that
173    contains the given object, or a null object if the given object is the last
174    in this list.
175-   `int mlirYIsNull(MlirY)` returns 1 if the given object is null.
176
177Note that the name of subobject in the function may or may not match its type.
178
179This approach enables one to iterate as follows.
180
181```c++
182MlirY iter;
183for (iter = mlirXGetFirst<Y>(x); !mlirYIsNull(iter);
184     iter = mlirYGetNextIn<X>(iter)) {
185  /* User 'iter'. */
186}
187```
188
189## Extending the API
190
191### Extensions for Dialect Attributes and Types
192
193Dialect attributes and types can follow the example of standard attributes and
194types, provided that implementations live in separate directories, i.e.
195`include/mlir-c/<...>Dialect/` and `lib/CAPI/<...>Dialect/`. The core APIs
196provide implementation-private headers in `include/mlir/CAPI/IR` that allow one
197to convert between opaque C structures for core IR components and their C++
198counterparts. `wrap` converts a C++ class into a C structure and `unwrap` does
199the inverse conversion. Once the C++ object is available, the API
200implementation should rely on `isa` to implement `mlirXIsAY` and is expected to
201use `cast` inside other API calls.
202