xref: /aosp_15_r20/art/test/646-checker-simd-hadd/src/HaddByte.java (revision 795d594fd825385562da6b089ea9b2033f3abf5a)
1*795d594fSAndroid Build Coastguard Worker /*
2*795d594fSAndroid Build Coastguard Worker  * Copyright (C) 2017 The Android Open Source Project
3*795d594fSAndroid Build Coastguard Worker  *
4*795d594fSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*795d594fSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*795d594fSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*795d594fSAndroid Build Coastguard Worker  *
8*795d594fSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*795d594fSAndroid Build Coastguard Worker  *
10*795d594fSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*795d594fSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*795d594fSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*795d594fSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*795d594fSAndroid Build Coastguard Worker  * limitations under the License.
15*795d594fSAndroid Build Coastguard Worker  */
16*795d594fSAndroid Build Coastguard Worker 
17*795d594fSAndroid Build Coastguard Worker /**
18*795d594fSAndroid Build Coastguard Worker  * Tests for halving-add idiomatic vectorization.
19*795d594fSAndroid Build Coastguard Worker  */
20*795d594fSAndroid Build Coastguard Worker public class HaddByte {
21*795d594fSAndroid Build Coastguard Worker 
22*795d594fSAndroid Build Coastguard Worker   private static final int N = 256;
23*795d594fSAndroid Build Coastguard Worker   private static final int M = N * N + 15;
24*795d594fSAndroid Build Coastguard Worker 
25*795d594fSAndroid Build Coastguard Worker   static byte[] sB1 = new byte[M];
26*795d594fSAndroid Build Coastguard Worker   static byte[] sB2 = new byte[M];
27*795d594fSAndroid Build Coastguard Worker   static byte[] sBo = new byte[M];
28*795d594fSAndroid Build Coastguard Worker 
29*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_signed(byte[], byte[], byte[]) loop_optimization (before)
30*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
31*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
32*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
33*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
34*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<Get1>>,<<Get2>>]             loop:<<Loop>>      outer_loop:none
35*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
36*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
37*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
38*795d594fSAndroid Build Coastguard Worker   //
39*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.halving_add_signed(byte[], byte[], byte[]) loop_optimization (after)
40*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
41*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
42*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Int8 rounded:false loop:<<Loop>> outer_loop:none
43*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
44*795d594fSAndroid Build Coastguard Worker   //
45*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.halving_add_signed(byte[], byte[], byte[]) loop_optimization (after)
46*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
47*795d594fSAndroid Build Coastguard Worker   //
48*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
49*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
50*795d594fSAndroid Build Coastguard Worker   //
51*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
52*795d594fSAndroid Build Coastguard Worker   //
53*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
54*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
55*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Int8 rounded:false loop:<<Loop>> outer_loop:none
56*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
57*795d594fSAndroid Build Coastguard Worker   //
58*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
halving_add_signed(byte[] b1, byte[] b2, byte[] bo)59*795d594fSAndroid Build Coastguard Worker   private static void halving_add_signed(byte[] b1, byte[] b2, byte[] bo) {
60*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, Math.min(b1.length, b2.length));
61*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
62*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) ((b1[i] + b2[i]) >> 1);
63*795d594fSAndroid Build Coastguard Worker     }
64*795d594fSAndroid Build Coastguard Worker   }
65*795d594fSAndroid Build Coastguard Worker 
66*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_unsigned(byte[], byte[], byte[]) instruction_simplifier (before)
67*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
68*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I255:i\d+>> IntConstant 255                     loop:none
69*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
70*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
71*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
72*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<And1:i\d+>> And [<<Get1>>,<<I255>>]             loop:<<Loop>>      outer_loop:none
73*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<And2:i\d+>> And [<<Get2>>,<<I255>>]             loop:<<Loop>>      outer_loop:none
74*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<And1>>,<<And2>>]             loop:<<Loop>>      outer_loop:none
75*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
76*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
77*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},{{i\d+}},<<Cnv>>] loop:<<Loop>>      outer_loop:none
78*795d594fSAndroid Build Coastguard Worker   //
79*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (before)
80*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
81*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
82*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:a\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
83*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:a\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
84*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<Get1>>,<<Get2>>]             loop:<<Loop>>      outer_loop:none
85*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
86*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
87*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
88*795d594fSAndroid Build Coastguard Worker   //
89*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (after)
90*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
91*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
92*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Uint8 rounded:false loop:<<Loop>> outer_loop:none
93*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
94*795d594fSAndroid Build Coastguard Worker   //
95*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (after)
96*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
97*795d594fSAndroid Build Coastguard Worker   //
98*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
99*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
100*795d594fSAndroid Build Coastguard Worker   //
101*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
102*795d594fSAndroid Build Coastguard Worker   //
103*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
104*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
105*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Uint8 rounded:false loop:<<Loop>> outer_loop:none
106*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
107*795d594fSAndroid Build Coastguard Worker   //
108*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
halving_add_unsigned(byte[] b1, byte[] b2, byte[] bo)109*795d594fSAndroid Build Coastguard Worker   private static void halving_add_unsigned(byte[] b1, byte[] b2, byte[] bo) {
110*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, Math.min(b1.length, b2.length));
111*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
112*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) (((b1[i] & 0xff) + (b2[i] & 0xff)) >> 1);
113*795d594fSAndroid Build Coastguard Worker     }
114*795d594fSAndroid Build Coastguard Worker   }
115*795d594fSAndroid Build Coastguard Worker 
116*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.rounding_halving_add_signed(byte[], byte[], byte[]) loop_optimization (before)
117*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
118*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
119*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
120*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
121*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add1:i\d+>> Add [<<Get1>>,<<Get2>>]             loop:<<Loop>>      outer_loop:none
122*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add2:i\d+>> Add [<<Add1>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
123*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add2>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
124*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
125*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
126*795d594fSAndroid Build Coastguard Worker   //
127*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.rounding_halving_add_signed(byte[], byte[], byte[]) loop_optimization (after)
128*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
129*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
130*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Int8 rounded:true loop:<<Loop>> outer_loop:none
131*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
132*795d594fSAndroid Build Coastguard Worker   //
133*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.rounding_halving_add_signed(byte[], byte[], byte[]) loop_optimization (after)
134*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
135*795d594fSAndroid Build Coastguard Worker   //
136*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
137*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
138*795d594fSAndroid Build Coastguard Worker   //
139*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
140*795d594fSAndroid Build Coastguard Worker   //
141*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
142*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
143*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>] packed_type:Int8 rounded:true loop:<<Loop>> outer_loop:none
144*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
145*795d594fSAndroid Build Coastguard Worker   //
146*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
rounding_halving_add_signed(byte[] b1, byte[] b2, byte[] bo)147*795d594fSAndroid Build Coastguard Worker   private static void rounding_halving_add_signed(byte[] b1, byte[] b2, byte[] bo) {
148*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, Math.min(b1.length, b2.length));
149*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
150*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) ((b1[i] + b2[i] + 1) >> 1);
151*795d594fSAndroid Build Coastguard Worker     }
152*795d594fSAndroid Build Coastguard Worker   }
153*795d594fSAndroid Build Coastguard Worker 
154*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.rounding_halving_add_unsigned(byte[], byte[], byte[]) instruction_simplifier (before)
155*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
156*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I255:i\d+>> IntConstant 255                     loop:none
157*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
158*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
159*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:b\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
160*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<And1:i\d+>> And [<<Get1>>,<<I255>>]             loop:<<Loop>>      outer_loop:none
161*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<And2:i\d+>> And [<<Get2>>,<<I255>>]             loop:<<Loop>>      outer_loop:none
162*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add1:i\d+>> Add [<<And1>>,<<And2>>]             loop:<<Loop>>      outer_loop:none
163*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add2:i\d+>> Add [<<Add1>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
164*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add2>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
165*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
166*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},{{i\d+}},<<Cnv>>] loop:<<Loop>>      outer_loop:none
167*795d594fSAndroid Build Coastguard Worker   //
168*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.rounding_halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (before)
169*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
170*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
171*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:a\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
172*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:a\d+>> ArrayGet                            loop:<<Loop>>      outer_loop:none
173*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add1:i\d+>> Add [<<Get1>>,<<Get2>>]             loop:<<Loop>>      outer_loop:none
174*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add2:i\d+>> Add [<<Add1>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
175*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add2>>,<<I1>>]               loop:<<Loop>>      outer_loop:none
176*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
177*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
178*795d594fSAndroid Build Coastguard Worker   //
179*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.rounding_halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (after)
180*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
181*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
182*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>]  packed_type:Uint8 rounded:true loop:<<Loop>> outer_loop:none
183*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
184*795d594fSAndroid Build Coastguard Worker   //
185*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.rounding_halving_add_unsigned(byte[], byte[], byte[]) loop_optimization (after)
186*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
187*795d594fSAndroid Build Coastguard Worker   //
188*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
189*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
190*795d594fSAndroid Build Coastguard Worker   //
191*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
192*795d594fSAndroid Build Coastguard Worker   //
193*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get1:d\d+>> VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
194*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get2:d\d+>> VecLoad                               loop:<<Loop>>      outer_loop:none
195*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get1>>,<<Get2>>]  packed_type:Uint8 rounded:true loop:<<Loop>> outer_loop:none
196*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
197*795d594fSAndroid Build Coastguard Worker   //
198*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
rounding_halving_add_unsigned(byte[] b1, byte[] b2, byte[] bo)199*795d594fSAndroid Build Coastguard Worker   private static void rounding_halving_add_unsigned(byte[] b1, byte[] b2, byte[] bo) {
200*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, Math.min(b1.length, b2.length));
201*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
202*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) (((b1[i] & 0xff) + (b2[i] & 0xff) + 1) >> 1);
203*795d594fSAndroid Build Coastguard Worker     }
204*795d594fSAndroid Build Coastguard Worker   }
205*795d594fSAndroid Build Coastguard Worker 
206*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_signed_constant(byte[], byte[]) loop_optimization (before)
207*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
208*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I127:i\d+>> IntConstant 127                     loop:none
209*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
210*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get:b\d+>>  ArrayGet                            loop:<<Loop>>      outer_loop:none
211*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<Get>>,<<I127>>]              loop:<<Loop>>      outer_loop:none
212*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
213*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
214*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
215*795d594fSAndroid Build Coastguard Worker   //
216*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.halving_add_signed_constant(byte[], byte[]) loop_optimization (after)
217*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I127:i\d+>> IntConstant 127                       loop:none
218*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Repl:d\d+>> VecReplicateScalar [<<I127>>]         loop:none
219*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get:d\d+>>  VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
220*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get>>,<<Repl>>] packed_type:Int8 rounded:false loop:<<Loop>> outer_loop:none
221*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
222*795d594fSAndroid Build Coastguard Worker   //
223*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.halving_add_signed_constant(byte[], byte[]) loop_optimization (after)
224*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
225*795d594fSAndroid Build Coastguard Worker   //
226*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
227*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
228*795d594fSAndroid Build Coastguard Worker   //
229*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
230*795d594fSAndroid Build Coastguard Worker   //
231*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<I127:i\d+>> IntConstant 127                       loop:none
232*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Repl:d\d+>> VecReplicateScalar [<<I127>>]         loop:none
233*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get:d\d+>>  VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
234*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get>>,<<Repl>>] packed_type:Int8 rounded:false loop:<<Loop>> outer_loop:none
235*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
236*795d594fSAndroid Build Coastguard Worker   //
237*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
halving_add_signed_constant(byte[] b1, byte[] bo)238*795d594fSAndroid Build Coastguard Worker   private static void halving_add_signed_constant(byte[] b1, byte[] bo) {
239*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, b1.length);
240*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
241*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) ((b1[i] + 0x7f) >> 1);
242*795d594fSAndroid Build Coastguard Worker     }
243*795d594fSAndroid Build Coastguard Worker   }
244*795d594fSAndroid Build Coastguard Worker 
245*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_unsigned_constant(byte[], byte[]) instruction_simplifier (before)
246*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
247*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I255:i\d+>> IntConstant 255                     loop:none
248*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
249*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get:b\d+>>  ArrayGet                            loop:<<Loop>>      outer_loop:none
250*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<And:i\d+>>  And [<<Get>>,<<I255>>]              loop:<<Loop>>      outer_loop:none
251*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<And>>,<<I255>>]              loop:<<Loop>>      outer_loop:none
252*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
253*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
254*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},{{i\d+}},<<Cnv>>] loop:<<Loop>>      outer_loop:none
255*795d594fSAndroid Build Coastguard Worker   //
256*795d594fSAndroid Build Coastguard Worker   /// CHECK-START: void HaddByte.halving_add_unsigned_constant(byte[], byte[]) loop_optimization (before)
257*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I1:i\d+>>   IntConstant 1                       loop:none
258*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I255:i\d+>> IntConstant 255                     loop:none
259*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Phi:i\d+>>  Phi                                 loop:<<Loop:B\d+>> outer_loop:none
260*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get:a\d+>>  ArrayGet                            loop:<<Loop>>      outer_loop:none
261*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Add:i\d+>>  Add [<<Get>>,<<I255>>]              loop:<<Loop>>      outer_loop:none
262*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Shr:i\d+>>  Shr [<<Add>>,<<I1>>]                loop:<<Loop>>      outer_loop:none
263*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Cnv:b\d+>>  TypeConversion [<<Shr>>]            loop:<<Loop>>      outer_loop:none
264*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               ArraySet [{{l\d+}},<<Phi>>,<<Cnv>>] loop:<<Loop>>      outer_loop:none
265*795d594fSAndroid Build Coastguard Worker   //
266*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM: void HaddByte.halving_add_unsigned_constant(byte[], byte[]) loop_optimization (after)
267*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<I255:i\d+>> IntConstant 255                       loop:none
268*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Repl:d\d+>> VecReplicateScalar [<<I255>>]         loop:none
269*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<Get:d\d+>>  VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
270*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get>>,<<Repl>>] packed_type:Uint8 rounded:false loop:<<Loop>> outer_loop:none
271*795d594fSAndroid Build Coastguard Worker   /// CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
272*795d594fSAndroid Build Coastguard Worker   //
273*795d594fSAndroid Build Coastguard Worker   /// CHECK-START-ARM64: void HaddByte.halving_add_unsigned_constant(byte[], byte[]) loop_optimization (after)
274*795d594fSAndroid Build Coastguard Worker   /// CHECK-IF:     hasIsaFeature("sve") and os.environ.get('ART_FORCE_TRY_PREDICATED_SIMD') == 'true'
275*795d594fSAndroid Build Coastguard Worker   //
276*795d594fSAndroid Build Coastguard Worker   //      HalvingAdd idiom is not supported for SVE.
277*795d594fSAndroid Build Coastguard Worker   ///     CHECK-NOT: VecHalvingAdd
278*795d594fSAndroid Build Coastguard Worker   //
279*795d594fSAndroid Build Coastguard Worker   /// CHECK-ELSE:
280*795d594fSAndroid Build Coastguard Worker   //
281*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<I255:i\d+>> IntConstant 255                       loop:none
282*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Repl:d\d+>> VecReplicateScalar [<<I255>>]         loop:none
283*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<Get:d\d+>>  VecLoad                               loop:<<Loop:B\d+>> outer_loop:none
284*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG: <<HAdd:d\d+>> VecHalvingAdd [<<Get>>,<<Repl>>] packed_type:Uint8 rounded:false loop:<<Loop>> outer_loop:none
285*795d594fSAndroid Build Coastguard Worker   ///     CHECK-DAG:               VecStore [{{l\d+}},{{i\d+}},<<HAdd>>] loop:<<Loop>>      outer_loop:none
286*795d594fSAndroid Build Coastguard Worker   //
287*795d594fSAndroid Build Coastguard Worker   /// CHECK-FI:
halving_add_unsigned_constant(byte[] b1, byte[] bo)288*795d594fSAndroid Build Coastguard Worker   private static void halving_add_unsigned_constant(byte[] b1, byte[] bo) {
289*795d594fSAndroid Build Coastguard Worker     int min_length = Math.min(bo.length, b1.length);
290*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < min_length; i++) {
291*795d594fSAndroid Build Coastguard Worker       bo[i] = (byte) (((b1[i] & 0xff) + 0xff) >> 1);
292*795d594fSAndroid Build Coastguard Worker     }
293*795d594fSAndroid Build Coastguard Worker   }
294*795d594fSAndroid Build Coastguard Worker 
main()295*795d594fSAndroid Build Coastguard Worker   public static void main() {
296*795d594fSAndroid Build Coastguard Worker     // Initialize cross-values to test all cases, and also
297*795d594fSAndroid Build Coastguard Worker     // set up some extra values to exercise the cleanup loop.
298*795d594fSAndroid Build Coastguard Worker     int k = 0;
299*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < N; i++) {
300*795d594fSAndroid Build Coastguard Worker       for (int j = 0; j < N; j++) {
301*795d594fSAndroid Build Coastguard Worker         sB1[k] = (byte) i;
302*795d594fSAndroid Build Coastguard Worker         sB2[k] = (byte) j;
303*795d594fSAndroid Build Coastguard Worker         k++;
304*795d594fSAndroid Build Coastguard Worker       }
305*795d594fSAndroid Build Coastguard Worker     }
306*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < 15; i++) {
307*795d594fSAndroid Build Coastguard Worker       sB1[k] = (byte) i;
308*795d594fSAndroid Build Coastguard Worker       sB2[k] = 100;
309*795d594fSAndroid Build Coastguard Worker       k++;
310*795d594fSAndroid Build Coastguard Worker     }
311*795d594fSAndroid Build Coastguard Worker     expectEquals(k, M);
312*795d594fSAndroid Build Coastguard Worker 
313*795d594fSAndroid Build Coastguard Worker     // Test halving add idioms.
314*795d594fSAndroid Build Coastguard Worker     halving_add_signed(sB1, sB2, sBo);
315*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
316*795d594fSAndroid Build Coastguard Worker       byte e = (byte) ((sB1[i] + sB2[i]) >> 1);
317*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
318*795d594fSAndroid Build Coastguard Worker     }
319*795d594fSAndroid Build Coastguard Worker     halving_add_unsigned(sB1, sB2, sBo);
320*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
321*795d594fSAndroid Build Coastguard Worker       byte e = (byte) (((sB1[i] & 0xff) + (sB2[i] & 0xff)) >> 1);
322*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
323*795d594fSAndroid Build Coastguard Worker     }
324*795d594fSAndroid Build Coastguard Worker     rounding_halving_add_signed(sB1, sB2, sBo);
325*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
326*795d594fSAndroid Build Coastguard Worker       byte e = (byte) ((sB1[i] + sB2[i] + 1) >> 1);
327*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
328*795d594fSAndroid Build Coastguard Worker     }
329*795d594fSAndroid Build Coastguard Worker     rounding_halving_add_unsigned(sB1, sB2, sBo);
330*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
331*795d594fSAndroid Build Coastguard Worker       byte e = (byte) (((sB1[i] & 0xff) + (sB2[i] & 0xff) + 1) >> 1);
332*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
333*795d594fSAndroid Build Coastguard Worker     }
334*795d594fSAndroid Build Coastguard Worker     halving_add_signed_constant(sB1, sBo);
335*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
336*795d594fSAndroid Build Coastguard Worker       byte e = (byte) ((sB1[i] + 0x7f) >> 1);
337*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
338*795d594fSAndroid Build Coastguard Worker     }
339*795d594fSAndroid Build Coastguard Worker     halving_add_unsigned_constant(sB1, sBo);
340*795d594fSAndroid Build Coastguard Worker     for (int i = 0; i < M; i++) {
341*795d594fSAndroid Build Coastguard Worker       byte e = (byte) (((sB1[i] & 0xff) + 0xff) >> 1);
342*795d594fSAndroid Build Coastguard Worker       expectEquals(e, sBo[i]);
343*795d594fSAndroid Build Coastguard Worker     }
344*795d594fSAndroid Build Coastguard Worker 
345*795d594fSAndroid Build Coastguard Worker     System.out.println("HaddByte passed");
346*795d594fSAndroid Build Coastguard Worker   }
347*795d594fSAndroid Build Coastguard Worker 
expectEquals(int expected, int result)348*795d594fSAndroid Build Coastguard Worker   private static void expectEquals(int expected, int result) {
349*795d594fSAndroid Build Coastguard Worker     if (expected != result) {
350*795d594fSAndroid Build Coastguard Worker       throw new Error("Expected: " + expected + ", found: " + result);
351*795d594fSAndroid Build Coastguard Worker     }
352*795d594fSAndroid Build Coastguard Worker   }
353*795d594fSAndroid Build Coastguard Worker }
354