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