20 lines
561 B
MLIR
20 lines
561 B
MLIR
// 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>
|
|
}
|