16 lines · plain
1// RUN: mlir-opt -inline %s | FileCheck %s2 3shard.grid @grid0(shape = 4x?x2)4 5func.func private @grid_to_inline() -> (index, index) {6 %0:2 = shard.grid_shape @grid0 axes = [2, 1] : index, index7 return %0#0, %0#1 : index, index8}9// CHECK-LABEL: func.func @main10func.func @main() -> (index, index) {11 // CHECK-NEXT: %[[AXIS_SIZE:.*]]:2 = shard.grid_shape @grid0 axes = [2, 1] : index12 %0:2 = func.call @grid_to_inline() : () -> (index, index)13 // CHECK-NEXT: return %[[AXIS_SIZE]]#0, %[[AXIS_SIZE]]#114 return %0#0, %0#1 : index, index15}16