| // RUN: mlir-opt %s -inline | FileCheck %s |
| |
| // Verifies that regions with operations from the ml_program dialect can |
| // be inlined. |
| |
| ml_program.global private @global(dense<4> : tensor<4xi32>) : tensor<4xi32> |
| |
| // CHECK: @inline_into |
| func.func @inline_into() -> tensor<4xi32> { |
| // CHECK-NOT: @inline_from |
| // CHECK: ml_program.global_load_const |
| %0 = call @inline_from() : () -> tensor<4xi32> |
| return %0 : tensor<4xi32> |
| } |
| |
| func.func @inline_from() -> tensor<4xi32> { |
| %0 = ml_program.global_load_const @global : tensor<4xi32> |
| return %0 : tensor<4xi32> |
| } |