Lines Matching +full:matrix +full:- +full:vector
1 // RUN: mlir-opt %s -test-lower-to-llvm | \
2 // RUN: mlir-runner -e entry -entry-point-result=void \
3 // RUN: -O0 -enable-matrix -matrix-allow-contract -matrix-default-layout=column-major \
4 // RUN: -shared-libs=%mlir_c_runner_utils | \
18 %0 = vector.broadcast %f0 : f64 to vector<4xf64>
19 %1 = vector.insert %f1, %0[1] : f64 into vector<4xf64>
20 %2 = vector.insert %f2, %1[2] : f64 into vector<4xf64>
21 %a = vector.insert %f3, %2[3] : f64 into vector<4xf64>
22 %3 = vector.broadcast %f4 : f64 to vector<4xf64>
23 %4 = vector.insert %f5, %3[1] : f64 into vector<4xf64>
24 %5 = vector.insert %f6, %4[2] : f64 into vector<4xf64>
25 %b = vector.insert %f7, %5[3] : f64 into vector<4xf64>
26 %6 = vector.broadcast %f0 : f64 to vector<6xf64>
27 %7 = vector.insert %f1, %6[1] : f64 into vector<6xf64>
28 %8 = vector.insert %f2, %7[2] : f64 into vector<6xf64>
29 %9 = vector.insert %f3, %8[3] : f64 into vector<6xf64>
30 %10 = vector.insert %f4, %9[4] : f64 into vector<6xf64>
31 %c = vector.insert %f5, %10[5] : f64 into vector<6xf64>
33 vector.print %a : vector<4xf64>
34 vector.print %b : vector<4xf64>
35 vector.print %c : vector<6xf64>
43 // Performs matrix transpositions interpreting the vectors as
44 // flattened column-major 2-D matrices.
47 // ( 1, 3 ) -> ( 2, 3 ) |/ | column-major!
50 // ( 5, 7 ) -> ( 6, 7 )
53 // ( 1, 3, 5 ) -> ( 2, 3 )
57 // ( 1, 4 ) -> ( 3, 4, 5 )
60 %d = vector.flat_transpose %a { rows = 2: i32, columns = 2: i32 } : vector<4xf64> -> vector<4xf64>
61 %e = vector.flat_transpose %b { rows = 2: i32, columns = 2: i32 } : vector<4xf64> -> vector<4xf64>
62 %f = vector.flat_transpose %c { rows = 2: i32, columns = 3: i32 } : vector<6xf64> -> vector<6xf64>
63 %g = vector.flat_transpose %c { rows = 3: i32, columns = 2: i32 } : vector<6xf64> -> vector<6xf64>
65 vector.print %d : vector<4xf64>
66 vector.print %e : vector<4xf64>
67 vector.print %f : vector<6xf64>
68 vector.print %g : vector<6xf64>