|
4 | 4 | // RUN: | FileCheck %s
|
5 | 5 |
|
6 | 6 | func @main() {
|
7 |
| - %A = constant dense<[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]> : tensor<2x3xf32> |
| 7 | + %A = constant dense<[[1.0, 2.0], [4.0, 5.0]]> : tensor<2x2xf32> |
8 | 8 | %B = constant dense<[[1.0, 2.0, 3.0, 4.0],
|
9 |
| - [5.0, 6.0, 7.0, 8.0], |
10 |
| - [9.0, 10.0, 11.0, 12.0]]> : tensor<3x4xf32> |
| 9 | + [5.0, 6.0, 7.0, 8.0]]> : tensor<2x4xf32> |
11 | 10 | %C = constant dense<1000.0> : tensor<2x4xf32>
|
12 | 11 |
|
13 |
| - %D = linalg.matmul ins(%A, %B: tensor<2x3xf32>, tensor<3x4xf32>) |
| 12 | + %D = linalg.matmul ins(%A, %B: tensor<2x2xf32>, tensor<2x4xf32>) |
14 | 13 | init(%C: tensor<2x4xf32>) -> tensor<2x4xf32>
|
15 | 14 |
|
16 | 15 | %unranked = tensor_cast %D : tensor<2x4xf32> to tensor<*xf32>
|
17 | 16 | call @print_memref_f32(%unranked) : (tensor<*xf32>) -> ()
|
18 | 17 |
|
19 | 18 | // CHECK: Unranked Memref base@ = {{0x[-9a-f]*}}
|
20 | 19 | // CHECK-SAME: rank = 2 offset = 0 sizes = [2, 4] strides = [4, 1] data =
|
21 |
| - // CHECK-NEXT: [1038, 1044, 1050, 1056] |
22 |
| - // CHECK-NEXT: [1065, 1074, 1083, 1092] |
| 20 | + // CHECK-NEXT: [1011, 1014, 1017, 1020] |
| 21 | + // CHECK-NEXT: [1029, 1038, 1047, 1056] |
23 | 22 |
|
24 | 23 | return
|
25 | 24 | }
|
|
0 commit comments