xref: /llvm-project/mlir/include/mlir/Dialect/MLProgram/IR/MLProgramBase.td (revision 863c346209e27d22157fad21d0fd730e710a3441)
1//===- MLProgramBase.td - Base defs for ml_program dialect --*- tablegen -*-==//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef MLPROGRAM_BASE
10#define MLPROGRAM_BASE
11
12include "mlir/IR/OpBase.td"
13
14def MLProgram_Dialect : Dialect {
15  let name = "ml_program";
16  let cppNamespace = "::mlir::ml_program";
17  let description = [{
18    The MLProgram dialect contains structural operations and types for
19    defining a compiled Machine-Learning program, as created from common
20    ML frameworks, such as TensorFlow, PyTorch, JAX, etc. It does not itself
21    define computation ops common to such frameworks but establishes a common
22    programming model for establishing modules, functions, globals and
23    memory model components appropriate for such an abstract level of detail.
24
25    This dialect is under active development, and while stability is an
26    eventual goal, it is not guaranteed at this juncture. Given the early state,
27    it is recommended to inquire further prior to using this dialect.
28  }];
29
30  let useDefaultAttributePrinterParser = 1;
31  let useDefaultTypePrinterParser = 1;
32}
33
34#endif // MLPROGRAM_BASE
35