Skip to content

Commit 45e5b8b

Browse files
committed
add in code to get nnz of input tensor
1 parent ba7bb6f commit 45e5b8b

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

numpy/minmax.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,16 @@ def bench():
2323
reduced = np.min(reduced, -1)
2424
return reduced
2525
tacoBench(bench)
26+
27+
@pytest.mark.parametrize("dims", [1, 3, 5])
28+
def bench_minmax_statistics(tacoBench, dims):
29+
loader = MinMaxPydataSparseTensorLoader()
30+
dims_list = [20] + [20] + [43 for ele in range(dims)]
31+
matrix = loader.tensor(dims_list)
32+
33+
extra_info = dict()
34+
extra_info["nnz"] = matrix.nnz
35+
36+
def nop():
37+
return 0
38+
tacoBench(nop, extra_info)

0 commit comments

Comments
 (0)