diff options
Diffstat (limited to 'src/main/java/org/linaro/benchmarks/vector/TestSIMDAdd.java')
-rw-r--r-- | src/main/java/org/linaro/benchmarks/vector/TestSIMDAdd.java | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/src/main/java/org/linaro/benchmarks/vector/TestSIMDAdd.java b/src/main/java/org/linaro/benchmarks/vector/TestSIMDAdd.java new file mode 100644 index 0000000..a7c69b4 --- /dev/null +++ b/src/main/java/org/linaro/benchmarks/vector/TestSIMDAdd.java @@ -0,0 +1,155 @@ +/* + * Copyright (C) 2016 Linaro Limited. All rights received. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + */ +package org.linaro.benchmarks.vector; + +import org.openjdk.jmh.annotations.*; +import java.util.concurrent.TimeUnit; + +@BenchmarkMode(Mode.AverageTime) +@OutputTimeUnit(TimeUnit.MICROSECONDS) +@State(Scope.Benchmark) + +public class TestSIMDAdd { + static final int LENGTH = 1024; + static final int COUNT = 128; + static int [] a = new int[LENGTH]; + static int [] b = new int[LENGTH]; + static int [] c = new int[LENGTH]; + static short [] sa = new short[LENGTH]; + static short [] sb = new short[LENGTH]; + static short [] sc = new short[LENGTH]; + static byte [] ba = new byte[LENGTH]; + static byte [] bb = new byte[LENGTH]; + static byte [] bc = new byte[LENGTH]; + static long [] la = new long[LENGTH]; + static long [] lb = new long[LENGTH]; + static long [] lc = new long[LENGTH]; + static float [] fa = new float[LENGTH]; + static float [] fb = new float[LENGTH]; + static float [] fc = new float[LENGTH]; + static double [] da = new double[LENGTH]; + static double [] db = new double[LENGTH]; + static double [] dc = new double[LENGTH]; + + public static void TestSIMDAddInit() { + for (int i = 0; i < LENGTH; i++) { + a[i] = i + 3; + b[i] = i + 2; + c[i] = i + 1; + sa[i] = (short)(i + 3); + sb[i] = (short)(i + 2); + sc[i] = (short)(i + 1); + ba[i] = (byte)(i + 3); + bb[i] = (byte)(i + 2); + bc[i] = (byte)(i + 1); + la[i] = (long)(i + 3); + lb[i] = (long)(i + 2); + lc[i] = (long)(i + 1); + fa[i] = (float)(i + 3); + fb[i] = (float)(i + 2); + fc[i] = (float)(i + 1); + da[i] = (double)(i + 3); + db[i] = (double)(i + 2); + dc[i] = (double)(i + 1); + } + } + + public static void vectAddInt() { + for (int i = 0; i < LENGTH; i++) { + c[i] = a[i] + b[i]; + } + } + + public static void vectAddShort() { + for (int i = 0; i < LENGTH; i++) { + sc[i] = (short)(sa[i] + sb[i]); + } + } + + public static void vectAddByte() { + for (int i = 0; i < LENGTH; i++) { + bc[i] = (byte)(ba[i] + bb[i]); + } + } + + public static void vectAddLong() { + for (int i = 0; i < LENGTH; i++) { + lc[i] = la[i] + lb[i]; + } + } + + public static void vectAddFloat() { + for (int i = 0; i < LENGTH; i++) { + fc[i] = fa[i] + fb[i]; + } + } + + public static void vectAddDouble() { + for (int i = 0; i < LENGTH; i++) { + dc[i] = da[i] + db[i]; + } + } + + @Setup + public void setup() + { + TestSIMDAddInit(); + } + + @Benchmark + public void testVectAddByte() { + for (int i = 0; i < COUNT; i++) { + vectAddByte(); + } + } + + @Benchmark + public void testVectAddShort() { + for (int i = 0; i < COUNT; i++) { + vectAddShort(); + } + } + + @Benchmark + public void testVectAddInt() { + for (int i = 0; i < COUNT; i++) { + vectAddInt(); + } + } + + @Benchmark + public void testVectAddLong() { + for (int i = 0; i < COUNT; i++) { + vectAddLong(); + } + } + + @Benchmark + public void testVectAddFloat() { + for (int i = 0; i < COUNT; i++) { + vectAddFloat(); + } + } + + @Benchmark + public void testVectAddDouble() { + for (int i = 0; i < COUNT; i++) { + vectAddDouble(); + } + } + +} |