numpy: add some more windowing benches

Author: rohany

Makefile

@@ -2,7 +2,7 @@ BENCHES := ""
 BENCHFLAGS :=
 
 python-bench: numpy/*.py
-	pytest $(BENCHFLAGS) $(BENCHES)
+	pytest --ignore=taco $(BENCHFLAGS) $(BENCHES)
 
 taco-bench: taco/build/taco-bench
 ifeq ($(BENCHES),"")

numpy/windowing.py

@@ -9,6 +9,8 @@ def test_add_sparse_window(benchmark):
     def bench():
         x = matrix[1:(dim-1), 1:(dim-1)] 
         res = x + x
+        # Sanity check that this has a similar runtime as taco.
+        # res = matrix + matrix
     benchmark.pedantic(bench, iterations=10)
 
 def test_add_multiple_sparse_windows(benchmark):
@@ -19,3 +21,21 @@ def test_add_multiple_sparse_windows(benchmark):
     def bench():
         res = matrix1[1:(dim-1), 1:(dim-1)] + matrix2[1:(dim-1), 1:(dim-1)] + matrix1[0:(dim-2), 0:(dim-2)]
     benchmark.pedantic(bench, iterations=10)
+
+def test_add_sparse_strided_window(benchmark):
+    dim = 10000
+    matrix = random(dim, dim, format="csr")
+    def bench():
+        x = matrix[1:(dim-1):2, 1:(dim-1):2] 
+        res = x + x
+    benchmark.pedantic(bench, iterations=10)
+
+def test_add_sparse_index_set(benchmark):
+    dim = 10000
+    indexes = [i * 2 for i in range(0, dim//2)]
+    matrix = random(dim, dim, format="csr")
+    def bench():
+        x = matrix[:, indexes] 
+        res = x + x
+    benchmark.pedantic(bench, iterations=10)
+