| /llvm-project-15.0.7/utils/bazel/llvm-project-overlay/mlir/python/ |
| H A D | BUILD.bazel | 25 "mlir/dialects/_ods_common.py", 125 "mlir/dialects/builtin.py", 236 "mlir/dialects/arith.py", 363 "mlir/dialects/cf.py", 405 "mlir/dialects/math.py", 541 "mlir/dialects/pdl.py", 602 "mlir/dialects/quant.py", 644 "mlir/dialects/scf.py", 686 "mlir/dialects/shape.py", 728 "mlir/dialects/func.py", [all …]
|
| /llvm-project-15.0.7/mlir/python/ |
| H A D | CMakeLists.txt | 22 dialects/_ods_common.py 70 dialects/builtin.py 79 dialects/complex.py 87 dialects/cf.py 95 dialects/func.py 113 dialects/linalg/*.py 158 dialects/arith.py 167 dialects/memref.py 185 dialects/quant.py 193 dialects/pdl.py [all …]
|
| /llvm-project-15.0.7/mlir/tools/mlir-tblgen/ |
| H A D | DialectGen.cpp | 59 Optional<Dialect> tblgen::findDialectToGenerate(ArrayRef<Dialect> dialects) { in findDialectToGenerate() argument 60 if (dialects.empty()) { in findDialectToGenerate() 66 if (dialects.size() == 1 && selectedDialect.getNumOccurrences() == 0) in findDialectToGenerate() 67 return dialects.front(); in findDialectToGenerate() 75 const auto *dialectIt = llvm::find_if(dialects, [](const Dialect &dialect) { in findDialectToGenerate() 78 if (dialectIt == dialects.end()) { in findDialectToGenerate() 239 SmallVector<Dialect> dialects(dialectDefs.begin(), dialectDefs.end()); in emitDialectDecls() local 240 Optional<Dialect> dialect = findDialectToGenerate(dialects); in emitDialectDecls() 310 SmallVector<Dialect> dialects(dialectDefs.begin(), dialectDefs.end()); in emitDialectDefs() local 311 Optional<Dialect> dialect = findDialectToGenerate(dialects); in emitDialectDefs()
|
| /llvm-project-15.0.7/mlir/test/python/ir/ |
| H A D | dialects.py | 37 d = ctx.dialects.func 41 _ = ctx.dialects.not_existing 48 d = ctx.dialects["func"] 52 _ = ctx.dialects["not_existing"] 85 op1 = ctx.dialects.arith.AddFOp(input1, input2) 88 from mlir.dialects.arith import AddFOp
|
| H A D | blocks.py | 7 from mlir.dialects import builtin 8 from mlir.dialects import cf 9 from mlir.dialects import func
|
| /llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/ |
| H A D | emit_fill.py | 4 from mlir.dialects import builtin 5 from mlir.dialects import func 6 from mlir.dialects import linalg 8 from mlir.dialects.linalg.opdsl.lang import *
|
| H A D | emit_convolution.py | 4 from mlir.dialects import builtin 5 from mlir.dialects import func 6 from mlir.dialects import linalg 8 from mlir.dialects.linalg.opdsl.lang import *
|
| H A D | emit_pooling.py | 4 from mlir.dialects import builtin 5 from mlir.dialects import func 6 from mlir.dialects import linalg 8 from mlir.dialects.linalg.opdsl.lang import *
|
| H A D | emit_misc.py | 4 from mlir.dialects import builtin 5 from mlir.dialects import func 6 from mlir.dialects import linalg 8 from mlir.dialects.linalg.opdsl.lang import *
|
| H A D | emit_matmul.py | 4 from mlir.dialects import builtin 5 from mlir.dialects import func 6 from mlir.dialects import linalg 8 from mlir.dialects.linalg.opdsl.lang import *
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | DialectRegistry.h | 50 MutableArrayRef<Dialect *> dialects) const = 0; 75 virtual void apply(MLIRContext *context, DialectsT *...dialects) const = 0; 89 MutableArrayRef<Dialect *> dialects) const final { in apply() argument 92 static_cast<DialectsT *>(dialects[dialectIdx++])...}; in apply() 207 void apply(MLIRContext *context, DialectsT *...dialects) const final { in addExtension() 208 extensionFn(context, dialects...); in addExtension()
|
| /llvm-project-15.0.7/mlir/test/python/dialects/ |
| H A D | tensor.py | 4 import mlir.dialects.arith as arith 5 import mlir.dialects.func as func 6 import mlir.dialects.tensor as tensor
|
| H A D | transform_loop_ext.py | 4 from mlir.dialects import transform 5 from mlir.dialects import pdl 6 from mlir.dialects.transform import loop
|
| H A D | scf.py | 4 from mlir.dialects import arith 5 from mlir.dialects import func 6 from mlir.dialects import scf 7 from mlir.dialects import builtin
|
| H A D | vector.py | 4 import mlir.dialects.builtin as builtin 5 import mlir.dialects.func as func 6 import mlir.dialects.vector as vector
|
| H A D | func.py | 4 from mlir.dialects import arith 5 from mlir.dialects import builtin 6 from mlir.dialects import func
|
| H A D | async_dialect.py | 4 import mlir.dialects.async_dialect 5 import mlir.dialects.async_dialect.passes
|
| H A D | gpu.py | 4 import mlir.dialects.gpu 5 import mlir.dialects.gpu.passes
|
| H A D | math_dialect.py | 7 import mlir.dialects.func as func 8 import mlir.dialects.math as mlir_math
|
| H A D | complex_dialect.py | 7 import mlir.dialects.func as func 8 import mlir.dialects.complex as mlir_complex
|
| /llvm-project-15.0.7/mlir/benchmark/python/ |
| H A D | common.py | 5 from mlir.dialects import arith 6 from mlir.dialects import func 7 from mlir.dialects import memref 8 from mlir.dialects import scf
|
| /llvm-project-15.0.7/mlir/docs/Tutorials/ |
| H A D | CreatingADialect.md | 5 Public dialects are typically separated into at least 3 directories: 36 Operations in dialects are typically declared using the ODS format in 86 access to all dialects. This list is also linked into libMLIR.so. 99 new targets with the same sources. However, dialects that depend on 132 on their source and target dialects, but may also depend on other 133 dialects (e.g. MLIRFunc). Typically this dependence is specified 152 (e.g. mlir-opt) which should have access to all dialects. This list 161 dialects to generate compile-time and link-time dependencies, and it 162 is not necessary to explicitly depend on the dialects' IncGen targets.
|
| /llvm-project-15.0.7/mlir/test/python/dialects/linalg/ |
| H A D | ops.py | 4 from mlir.dialects import arith 5 from mlir.dialects import builtin 6 from mlir.dialects import func 7 from mlir.dialects import linalg 9 from mlir.dialects.linalg.opdsl.lang import *
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/ |
| H A D | test_elementwise_add_sparse_output.py | 10 from mlir.dialects import sparse_tensor as st 11 from mlir.dialects import builtin 12 from mlir.dialects.linalg.opdsl import lang as dsl
|
| H A D | test_SpMM.py | 12 from mlir.dialects import sparse_tensor as st 13 from mlir.dialects import builtin 14 from mlir.dialects import func 15 from mlir.dialects.linalg.opdsl import lang as dsl
|