1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Test hlfir.transpose operation parse, verify (no errors), and unparse
// RUN: fir-opt %s | fir-opt | FileCheck %s
// square matrix of known shape
func.func @transpose0(%arg0: !hlfir.expr<2x2xi32>) {
%res = hlfir.transpose %arg0 : (!hlfir.expr<2x2xi32>) -> !hlfir.expr<2x2xi32>
return
}
// CHECK-LABEL: func.func @transpose0
// CHECK: %[[ARG0:.*]]: !hlfir.expr<2x2xi32>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!hlfir.expr<2x2xi32>) -> !hlfir.expr<2x2xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
// rectangular matrix of known shape
func.func @transpose1(%arg0: !hlfir.expr<2x3xi32>) {
%res = hlfir.transpose %arg0 : (!hlfir.expr<2x3xi32>) -> !hlfir.expr<3x2xi32>
return
}
// CHECK-LABEL: func.func @transpose1
// CHECK: %[[ARG0:.*]]: !hlfir.expr<2x3xi32>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!hlfir.expr<2x3xi32>) -> !hlfir.expr<3x2xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
// matrix of assumed shape
func.func @transpose2(%arg0: !hlfir.expr<?x?xi32>) {
%res = hlfir.transpose %arg0 : (!hlfir.expr<?x?xi32>) -> !hlfir.expr<?x?xi32>
return
}
// CHECK-LABEL: func.func @transpose2
// CHECK: %[[ARG0:.*]]: !hlfir.expr<?x?xi32>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!hlfir.expr<?x?xi32>) -> !hlfir.expr<?x?xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
// matrix where only some dimensions are known #1
func.func @transpose3(%arg0: !hlfir.expr<?x2xi32>) {
%res = hlfir.transpose %arg0 : (!hlfir.expr<?x2xi32>) -> !hlfir.expr<2x?xi32>
return
}
// CHECK-LABEL: func.func @transpose3
// CHECK: %[[ARG0:.*]]: !hlfir.expr<?x2xi32>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!hlfir.expr<?x2xi32>) -> !hlfir.expr<2x?xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
// matrix where only some dimensions are known #2
func.func @transpose4(%arg0: !hlfir.expr<2x?xi32>) {
%res = hlfir.transpose %arg0 : (!hlfir.expr<2x?xi32>) -> !hlfir.expr<?x2xi32>
return
}
// CHECK-LABEL: func.func @transpose4
// CHECK: %[[ARG0:.*]]: !hlfir.expr<2x?xi32>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!hlfir.expr<2x?xi32>) -> !hlfir.expr<?x2xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
// matrix is a boxed array
func.func @transpose5(%arg0: !fir.box<!fir.array<1x2xi32>>) {
%res = hlfir.transpose %arg0 : (!fir.box<!fir.array<1x2xi32>>) -> !hlfir.expr<2x1xi32>
return
}
// CHECK-LABEL: func.func @transpose5
// CHECK: %[[ARG0:.*]]: !fir.box<!fir.array<1x2xi32>>
// CHECK-NEXT: %[[RES:.*]] = hlfir.transpose %[[ARG0]] : (!fir.box<!fir.array<1x2xi32>>) -> !hlfir.expr<2x1xi32>
// CHECK-NEXT: return
// CHECK-NEXT: }
|
