1// RUN: mlir-opt -inline %s | FileCheck %s 2 3mesh.mesh @mesh0(shape = 4x?x2) 4 5func.func private @mesh_to_inline() -> (index, index) { 6 %0:2 = mesh.mesh_shape @mesh0 axes = [2, 1] : index, index 7 return %0#0, %0#1 : index, index 8} 9// CHECK-LABEL: func.func @main 10func.func @main() -> (index, index) { 11 // CHECK-NEXT: %[[AXIS_SIZE:.*]]:2 = mesh.mesh_shape @mesh0 axes = [2, 1] : index 12 %0:2 = func.call @mesh_to_inline() : () -> (index, index) 13 // CHECK-NEXT: return %[[AXIS_SIZE]]#0, %[[AXIS_SIZE]]#1 14 return %0#0, %0#1 : index, index 15} 16