2
2
3
3
module {
4
4
func.func @remove_stick_2d () -> tensor <2 x3 xf32 > {
5
- %0 = " zhigh.StickifiedConstant" () {alignment = 4096 : i64 , layout = " 2D " , value = dense <[[0. , 1. , 2. ], [3. , 4. , 5. ]]> : tensor <2 x3 xf32 >} : () -> tensor <2 x3 xf16 , #zhigh.layout <{dataLayout = " 2D" }>>
5
+ %0 = " zhigh.StickifiedConstant" () {alignment = 4096 : i64 , value = dense <[[0. , 1. , 2. ], [3. , 4. , 5. ]]> : tensor <2 x3 xf32 >} : () -> tensor <2 x3 xf16 , #zhigh.layout <{dataLayout = " 2D" }>>
6
6
%1 = " zhigh.Unstick" (%0 ) : (tensor <2 x3 xf16 , #zhigh.layout <{dataLayout = " 2D" }>>) -> tensor <2 x3 xf32 >
7
7
return %1 : tensor <2 x3 xf32 >
8
8
}
@@ -12,7 +12,7 @@ module {
12
12
// CHECK-DAG: [[MAP_0_:#.+]] = affine_map<(d0, d1) -> (0, d1 floordiv 64, 0, d0 floordiv 32, d0 mod 32, d1 mod 64)>
13
13
// CHECK-LABEL: func @remove_stick_2d
14
14
// CHECK-SAME: () -> memref<2x3xf32> {
15
- // CHECK-DAG: [[VAR_0_:%.+]] = "zlow.stickifiedConstant"() {alignment = 4096 : i64, name = "constant_stickify_0", offset = 0 : i64, shape = [1, 1, 1, 1, 32, 64], value = dense<{{.}}[0.000000e+00, 1.000000e+00, 2.000000e+00], [3.000000e+00, 4.000000e+00, 5.000000e+00]{{.}}> : tensor<2x3xf32>} : () -> memref<2x3xf16, #map>
15
+ // CHECK-DAG: [[VAR_0_:%.+]] = "zlow.stickifiedConstant"() {alignment = 4096 : i64, layout = "2D", name = "constant_stickify_0", offset = 0 : i64, shape = [1, 1, 1, 1, 32, 64], value = dense<{{.}}[0.000000e+00, 1.000000e+00, 2.000000e+00], [3.000000e+00, 4.000000e+00, 5.000000e+00]{{.}}> : tensor<2x3xf32>} : () -> memref<2x3xf16, #map>
16
16
// CHECK-DAG: [[VAR_c2_:%.+]] = arith.constant 2 : index
17
17
// CHECK-DAG: [[VAR_c3_:%.+]] = arith.constant 3 : index
18
18
// CHECK-DAG: [[RES_:%.+]] = memref.alloc() {{.*}}: memref<2x3xf32>
0 commit comments