Merge pull request #7 from Quafadas/apply_simd

Apply simd

benchmark/src/Benchmark.scala

@@ -38,6 +38,14 @@ abstract class BLASBenchmark:
     return res;
   end randomDoubleArray
 
+  protected def randomBooleanArray(n: Int): Array[Boolean] =
+    val res = new Array[Boolean](n);
+
+    for i <- 0 until n do res(i) = rand.nextDouble() < 0.5;
+    end for
+    return res;
+  end randomBooleanArray
+
   protected def randomFloat(): Float =
     return rand.nextFloat();
 

benchmark/src/and.scala

@@ -0,0 +1,83 @@
+/*
+ * Copyright 2020, 2021, Ludovic Henry
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ * Please contact git@ludovic.dev or visit ludovic.dev if you need additional
+ * information or have any questions.
+ */
+
+package vecxt.benchmark
+
+import org.openjdk.jmh.annotations.*
+import org.openjdk.jmh.infra.Blackhole
+// import vecxt.Matrix.*
+import vecxt.BoundsCheck
+import scala.compiletime.uninitialized
+import vecxt.*
+import jdk.incubator.vector.VectorSpecies
+import jdk.incubator.vector.VectorOperators
+import jdk.incubator.vector.DoubleVector
+
+@State(Scope.Thread)
+class AndBooleanBenchmark extends BLASBenchmark:
+
+  @Param(Array("3", "128", "100000"))
+  var len: String = uninitialized;
+
+  var arr: Array[Boolean] = uninitialized
+  var arr2: Array[Boolean] = uninitialized
+
+  // format: off
+  @Setup(Level.Trial)
+  def setup: Unit =
+    arr = randomBooleanArray(len.toInt);
+    arr2 = randomBooleanArray(len.toInt);
+    ()
+  end setup
+
+  extension (vec: Array[Boolean])
+    inline def and2(thatIdx: Array[Boolean]) =
+      val result: Array[Boolean] = new Array[Boolean](vec.length)
+      var i = 0
+
+      while i < vec.length do
+        result(i) = vec(i) && thatIdx(i)
+        i += 1
+      end while
+      result
+  end extension
+
+  @Benchmark
+  def and_loop(bh: Blackhole) =
+    val r = arr.and2(arr2)
+    bh.consume(r);
+  end and_loop
+
+
+  @Benchmark
+  def and_loop_vec(bh: Blackhole) =
+    val r = arr && arr2
+    bh.consume(r);
+  end and_loop_vec
+
+end AndBooleanBenchmark
+
+
+

benchmark/src/countTrue.scala

@@ -36,44 +36,44 @@ import jdk.incubator.vector.VectorOperators
 import jdk.incubator.vector.DoubleVector
 
 @State(Scope.Thread)
-class IncrementBenchmark extends BLASBenchmark:
+class CountTrueBenchmark extends BLASBenchmark:
 
-  @Param(Array("3", "100", "100000"))
+  @Param(Array("3", "128", "100000"))
   var len: String = uninitialized;
 
-  var arr: Array[Double] = uninitialized
+  var arr: Array[Boolean] = uninitialized
 
   // format: off
   @Setup(Level.Trial)
   def setup: Unit =
-    arr = randomDoubleArray(len.toInt);
+    arr = randomBooleanArray(len.toInt);
     ()
   end setup
 
-  extension (vec: Array[Double])
-    inline def increments_loop: Array[Double] =
-      val out = new Array[Double](vec.length)
-      out(0) = vec(0)
-      var i = 1
+  extension (vec: Array[Boolean])
+    inline def countTrue2: Int =
+      var sum = 0
+      var i = 0
       while i < vec.length do
-        out(i) = vec(i) - vec(i - 1)
-        i = i + 1
+        if vec(i) then sum += 1
+        end if
+        i += 1
       end while
-      out
-    end increments_loop
+      sum
 
   end extension
 
   @Benchmark
-  def increment_normal(bh: Blackhole) =
-    val r = arr.increments_loop
+  def countTrue_loop(bh: Blackhole) =
+    val r = arr.countTrue2
     bh.consume(r);
-  end increment_normal
+  end countTrue_loop
+
 
   @Benchmark
-  def increment_vec(bh: Blackhole) =
-    val r = arr.increments
+  def countTrue_loop_vec(bh: Blackhole) =
+    val r = arr.countTrue
     bh.consume(r);
-  end increment_vec
+  end countTrue_loop_vec
+end CountTrueBenchmark
 
-end IncrementBenchmark

benchmark/src/logical.scala

@@ -0,0 +1,81 @@
+/*
+ * Copyright 2020, 2021, Ludovic Henry
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ * Please contact git@ludovic.dev or visit ludovic.dev if you need additional
+ * information or have any questions.
+ */
+
+package vecxt.benchmark
+
+import org.openjdk.jmh.annotations.*
+import org.openjdk.jmh.infra.Blackhole
+// import vecxt.Matrix.*
+import vecxt.BoundsCheck
+import scala.compiletime.uninitialized
+import vecxt.*
+import jdk.incubator.vector.VectorSpecies
+import jdk.incubator.vector.VectorOperators
+import jdk.incubator.vector.DoubleVector
+
+@State(Scope.Thread)
+class LogicalBenchmark extends BLASBenchmark:
+
+  @Param(Array("3", "128", "100000"))
+  var len: String = uninitialized;
+
+  var arr: Array[Double] = uninitialized
+
+
+  // format: off
+  @Setup(Level.Trial)
+  def setup: Unit =
+    arr = randomDoubleArray(len.toInt);    
+    ()
+  end setup
+
+  extension (vec: Array[Double])
+    inline def lte2(num: Double) =
+      val idx: Array[Boolean] = new Array[Boolean](vec.length)
+      var i = 0
+
+      while i < vec.length do
+        idx(i) = vec(i) <= num
+        i += 1
+      end while
+      idx
+  end extension
+
+  @Benchmark
+  def lte_vec(bh: Blackhole) =
+    val r = arr <= 4.0
+    bh.consume(r);
+  end lte_vec
+
+
+
+  @Benchmark
+  def lte_loop(bh: Blackhole) =
+    val r = arr.lte2(4.0)
+    bh.consume(r);
+  end lte_loop
+
+end LogicalBenchmark
+

benchmark/src/or.scala

@@ -0,0 +1,78 @@
+/*
+ * Copyright 2020, 2021, Ludovic Henry
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ * Please contact git@ludovic.dev or visit ludovic.dev if you need additional
+ * information or have any questions.
+ */
+
+package vecxt.benchmark
+
+import org.openjdk.jmh.annotations.*
+import org.openjdk.jmh.infra.Blackhole
+// import vecxt.Matrix.*
+import vecxt.BoundsCheck
+import scala.compiletime.uninitialized
+import vecxt.*
+import jdk.incubator.vector.VectorSpecies
+import jdk.incubator.vector.VectorOperators
+import jdk.incubator.vector.DoubleVector
+
+@State(Scope.Thread)
+class OrBooleanBenchmark extends BLASBenchmark:
+
+  @Param(Array("3", "128", "100000"))
+  var len: String = uninitialized;
+
+  var arr: Array[Boolean] = uninitialized
+  var arr2: Array[Boolean] = uninitialized
+
+  // format: off
+  @Setup(Level.Trial)
+  def setup: Unit =
+    arr = randomBooleanArray(len.toInt);
+    arr2 = randomBooleanArray(len.toInt);
+    ()
+  end setup
+
+  extension (vec: Array[Boolean])
+    inline def or2(thatIdx: Array[Boolean]) =
+      val result: Array[Boolean] = new Array[Boolean](vec.length)
+      var i = 0
+
+      while i < vec.length do
+        result(i) = vec(i) || thatIdx(i)
+        i += 1
+      end while
+      result
+  end extension
+
+  @Benchmark
+  def or_loop(bh: Blackhole) =
+    val r = arr.or2(arr2)
+    bh.consume(r);
+  end or_loop
+
+  @Benchmark
+  def or_vec(bh: Blackhole) =
+    val r = arr || arr2
+    bh.consume(r);
+  end or_vec
+end OrBooleanBenchmark

justfile

@@ -7,7 +7,7 @@ benchmark:
   mill benchmark.runJmh -jvmArgs --add-modules=jdk.incubator.vector -rf json
 
 benchmarkOnly:
-  mill benchmark.runJmh -jvmArgs --add-modules=jdk.incubator.vector -rf json vecxt.benchmark.AddScalarBenchmark
+  mill benchmark.runJmh -jvmArgs --add-modules=jdk.incubator.vector -rf json vecxt.benchmark.LogicalBenchmark
 
 setJvm:
   eval "$(cs java --jvm 21 --env)"

vecxt/js/src/package.scala

@@ -113,7 +113,7 @@ object extensions:
 
   extension (vec: NArray[Double])
 
-    inline def idxBoolean(index: js.Array[Boolean])(using inline boundsCheck: BoundsCheck.BoundsCheck) =
+    inline def apply(index: js.Array[Boolean])(using inline boundsCheck: BoundsCheck.BoundsCheck) =
       dimCheck(vec, index)
       val trues = index.countTrue
       val newVec = Float64Array(trues)
@@ -125,7 +125,7 @@ object extensions:
           j = 1 + j
       end for
       newVec
-    end idxBoolean
+    end apply
 
     def increments: Float64Array =
       val out = Float64Array(vec.length)