Home
last modified time | relevance | path

Searched refs:ml_program (Results 1 – 25 of 28) sorted by relevance

12

/llvm-project/mlir/test/Dialect/MLProgram/
H A Dinvalid.mlir3 ml_program.func @ssa_enforced(%arg0 : i32) -> i32 {
8 ml_program.return %0 : i32
12 ml_program.func @return_arity_match(%arg0 : i32) -> i32 {
14 ml_program.return %arg0, %arg0 : i32, i32
18 ml_program.func @return_type_match(%arg0 : i64) -> i32 {
20 ml_program.return %arg0 : i64
24 ml_program.subgraph @output_arity_match(%arg0 : i32) -> i32 {
26 ml_program.output %arg0, %arg0 : i32, i32
30 ml_program.subgraph @output_type_match(%arg0 : i64) -> i32 {
32 ml_program.output %arg0 : i64
[all …]
H A Dpipeline-globals.mlir4 ml_program.global private mutable @global_variable(dense<4> : tensor<4xi32>) : tensor<4xi32>
8 // CHECK: %[[LOAD:.+]] = ml_program.global_load @global_variable
9 // CHECK-NOT: ml_program.global_load @global_variable
10 %0 = ml_program.global_load @global_variable : tensor<4xi32>
11 %1 = ml_program.global_load @global_variable : tensor<4xi32>
16 // CHECK: ml_program.global_store @global_variable %[[DUMMY]]
17 ml_program.global_store @global_variable = %2 : tensor<4xi32>
24 ml_program.global private mutable @global_variable(dense<4> : tensor<4xi32>) : tensor<4xi32>
28 // CHECK: %[[LOAD:.+]] = ml_program.global_load @global_variable
29 %0 = ml_program.global_load @global_variable : tensor<4xi32>
[all …]
H A Dops.mlir4 // CHECK-LABEL: ml_program.func private @extern_func
5 ml_program.func private @extern_func(i32) -> i32
7 // CHECK-LABEL: ml_program.func @defined_func
8 ml_program.func @defined_func(%arg0 : i32) -> i32 {
9 ml_program.return %arg0 : i32
12 // CHECK-LABEL: ml_program.subgraph private @extern_subgraph
13 ml_program.subgraph private @extern_subgraph(i32) -> i32
15 // CHECK-LABEL: ml_program.subgraph @compute_subgraph
16 ml_program.subgraph @compute_subgraph(%arg0 : i32) -> i32 {
17 %token = ml_program.token
[all …]
H A Done-shot-bufferize.mlir4 ml_program.global private mutable @global(dense<0> : tensor<i64>) : tensor<i64>
19 %0 = ml_program.global_load @global : tensor<i64>
23 ml_program.global_store @global = %inserted : tensor<i64>
30 ml_program.global private mutable @global(dense<0> : tensor<i64>) : tensor<i64>
45 %0 = ml_program.global_load @global : tensor<i64>
46 %1 = ml_program.global_load @global : tensor<i64>
49 ml_program.global_store @global = %inserted : tensor<i64>
H A Dinlining.mlir3 // Verifies that regions with operations from the ml_program dialect can
6 ml_program.global private @global(dense<4> : tensor<4xi32>) : tensor<4xi32>
11 // CHECK: ml_program.global_load_const
17 %0 = ml_program.global_load_const @global : tensor<4xi32>
H A Dattrs.mlir3 // CHECK: #ml_program.extern : i32
5 value = #ml_program.extern : i32
/llvm-project/mlir/test/python/dialects/
H A Dml_program.py5 from mlir.dialects import ml_program, arith, builtin
22 f = ml_program.FuncOp(
28 ml_program.ReturnOp([block.arguments[0]])
44 ml_program.GlobalOp("symbol2", F32Type.get(), is_mutable=True)
46 ml_program.GlobalStoreOp(["symbol1", "symbol2"], cst)
/llvm-project/mlir/include/mlir/Dialect/MLProgram/IR/
H A DMLProgramOps.td47 ml_program.func private @some_extern(i32) -> i32
48 ml_program.func @compute(%arg0 : i32) -> i32 {
49 ml_program.return %arg0 : i32
122 ml_program.global @foobar(dense<4> : tensor<4xi32>) : tensor<?xi32>
125 ml_program.global mutable @foobar(#ml_program.extern<tensor<4xi32>>)
129 ml_program.global mutable @foobar : tensor<?xi32>
176 %0 = ml_program.global_load @foobar : tensor<?xi32>
223 %0 = ml_program.global_load_const @foobar : tensor<?xi32>
273 %0, %cstr = ml_program.global_load_graph @foobar
274 ordering (%token -> !ml_program.token) : tensor<?xi32>
[all …]
H A DMLProgramBase.td1 //===- MLProgramBase.td - Base defs for ml_program dialect --*- tablegen -*-==//
15 let name = "ml_program";
16 let cppNamespace = "::mlir::ml_program";
H A DCMakeLists.txt2 add_mlir_dialect(MLProgramOps ml_program)
/llvm-project/mlir/test/Conversion/TosaToMLProgram/
H A Dtosa-to-mlprogram.mlir4 …// CHECK: ml_program.global private mutable @var_x(dense<7.000000e+00> : tensor<1xf32>) : tensor<1…
7 // CHECK: ml_program.global_store @var_x = %arg0 : tensor<1xf32>
9 // CHECK: %[[LOAD:.+]] = ml_program.global_load @var_x : tensor<1xf32>
/llvm-project/mlir/include/mlir/Dialect/MLProgram/Transforms/
H A DPasses.td15 let summary = "Optimize `ml_program` global operations for read and store";
17 `ml_program`'s load and store operations can be optimized for
24 let constructor = "mlir::ml_program::createMLProgramPipelineGlobalsPass()";
H A DBufferizableOpInterfaceImpl.h15 namespace ml_program {
H A DPasses.h17 namespace ml_program {
/llvm-project/mlir/lib/Conversion/TosaToMLProgram/
H A DTosaToMLProgram.cpp29 auto newVariable = rewriter.create<mlir::ml_program::GlobalOp>( in matchAndRewrite()
47 auto newVariableWrite = rewriter.create<ml_program::GlobalStoreOp>( in matchAndRewrite()
62 auto newVariableRead = rewriter.create<ml_program::GlobalLoadOp>( in matchAndRewrite()
/llvm-project/mlir/lib/CAPI/Dialect/
H A DMLProgram.cpp13 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(MLProgram, ml_program,
14 mlir::ml_program::MLProgramDialect)
/llvm-project/mlir/lib/Dialect/MLProgram/IR/
H A DMLProgramDialect.cpp15 using namespace mlir::ml_program;
53 void ml_program::MLProgramDialect::initialize() { in initialize()
H A DMLProgramOps.cpp14 using namespace mlir::ml_program;
/llvm-project/mlir/test/IR/
H A Dinvalid-file-metadata.mlir62 // expected-error@+4 {{unknown 'resource' key 'unknown_entry' for dialect 'ml_program'}}
65 ml_program: {
/llvm-project/mlir/include/mlir/
H A DInitAllDialects.h132 ml_program::MLProgramDialect, in registerAllDialects()
178 ml_program::registerBufferizableOpInterfaceExternalModels(registry); in registerAllDialects()
H A DInitAllPasses.h85 ml_program::registerMLProgramPasses(); in registerAllPasses()
/llvm-project/mlir/lib/Dialect/MLProgram/Transforms/
H A DBufferizableOpInterfaceImpl.cpp17 using namespace mlir::ml_program;
20 namespace ml_program { namespace
H A DPipelineGlobalOps.cpp18 namespace ml_program { namespace
232 } // namespace ml_program
/llvm-project/mlir/include/mlir-c/Dialect/
H A DMLProgram.h19 MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(MLProgram, ml_program);
/llvm-project/mlir/test/mlir-opt/
H A Dcommandline.mlir26 // CHECK-SAME: ml_program

12