diff options
Diffstat (limited to 'src/main/java/org/linaro/benchmarks/vector/TestSIMDFma.java')
-rw-r--r-- | src/main/java/org/linaro/benchmarks/vector/TestSIMDFma.java | 223 |
1 files changed, 223 insertions, 0 deletions
diff --git a/src/main/java/org/linaro/benchmarks/vector/TestSIMDFma.java b/src/main/java/org/linaro/benchmarks/vector/TestSIMDFma.java new file mode 100644 index 0000000..336ea87 --- /dev/null +++ b/src/main/java/org/linaro/benchmarks/vector/TestSIMDFma.java @@ -0,0 +1,223 @@ +/* + * Copyright (C) 2017 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 TestSIMDFma { + static final int LENGTH = 1024; + static final int COUNT = 128; + 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 testSIMDFmaInit() { + for (int i = 0; i < LENGTH; i++) { + 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 vecFusedMulAddFloat( + float[] a, + float[] b, + float[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], b[i], c[i]); + } + } + + public static void vecFusedMulSub1Float( + float[] a, + float[] b, + float[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(-a[i], b[i], c[i]); + } + } + + public static void vecFusedMulSub2Float( + float[] a, + float[] b, + float[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], -b[i], c[i]); + } + } + + public static void vecNegFusedMulSubFloat( + float[] a, + float[] b, + float[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], b[i], -c[i]); + } + } + + public static void vecNegFusedMulAddFloat( + float[] a, + float[] b, + float[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = -(Math.fma(a[i], b[i], c[i])); + } + } + + public static void vecFusedMulAddDouble( + double[] a, + double[] b, + double[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], b[i], c[i]); + } + } + + public static void vecFusedMulSub1Double( + double[] a, + double[] b, + double[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(-a[i], b[i], c[i]); + } + } + + public static void vecFusedMulSub2Double( + double[] a, + double[] b, + double[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], -b[i], c[i]); + } + } + + public static void vecNegFusedMulSubDouble( + double[] a, + double[] b, + double[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = Math.fma(a[i], b[i], -c[i]); + } + } + + public static void vecNegFusedMulAddDouble( + double[] a, + double[] b, + double[] c, + int loop) { + for (int i = 0; i < loop; i++) { + c[i] = -(Math.fma(a[i], b[i], c[i])); + } + } + + @Setup + public void setup() + { + testSIMDFmaInit(); + } + + @Benchmark + public void testVecFusedMulAddFloat() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulAddFloat(fa, fb, fc, LENGTH); + } + } + + @Benchmark + public void testVecFusedMulSub1Float() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulSub1Float(fa, fb, fc, LENGTH); + } + } + + @Benchmark + public void testVecFusedMulSub2Float() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulSub2Float(fa, fb, fc, LENGTH); + } + } + + @Benchmark + public void testVecNegFusedMulAddFloat() { + for (int i = 0; i < COUNT; i++) { + vecNegFusedMulAddFloat(fa, fb, fc, LENGTH); + } + } + + @Benchmark + public void testVecNegFusedMulSubFloat() { + for (int i = 0; i < COUNT; i++) { + vecNegFusedMulSubFloat(fa, fb, fc, LENGTH); + } + } + + @Benchmark + public void testVecFusedMulAddDouble() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulAddDouble(da, db, dc, LENGTH); + } + } + + @Benchmark + public void testVecFusedMulSub1Double() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulSub1Double(da, db, dc, LENGTH); + } + } + + @Benchmark + public void testVecFusedMulSub2Double() { + for (int i = 0; i < COUNT; i++) { + vecFusedMulSub2Double(da, db, dc, LENGTH); + } + } + + @Benchmark + public void testVecNegFusedMulAddDouble() { + for (int i = 0; i < COUNT; i++) { + vecNegFusedMulAddDouble(da, db, dc, LENGTH); + } + } + + @Benchmark + public void testVecNegFusedMulSubDouble() { + for (int i = 0; i < COUNT; i++) { + vecNegFusedMulSubDouble(da, db, dc, LENGTH); + } + } + +} |