rand FFI

apache · Jul 4, 2020 · 7b019ef · 7b019ef
1 parent b26fca5
commit 7b019ef
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/benchmark/python/ffi/benchmark_ffi.py b/benchmark/python/ffi/benchmark_ffi.py
@@ -120,6 +120,7 @@ def prepare_workloads():
     OpArgMngr.add_workload("diff", pool['2x2'], n=1, axis=-1)
     OpArgMngr.add_workload("inner", pool['2x2'], pool['2x2'])
     OpArgMngr.add_workload("random.multinomial", n=2, pvals=[1/6.]*6, size=(2,2))
+    OpArgMngr.add_workload("random.rand", 3, 2)
     OpArgMngr.add_workload("nonzero", pool['2x2'])
     OpArgMngr.add_workload("tril", pool['2x2'], k=0)
     OpArgMngr.add_workload("random.choice", pool['2'], size=(2, 2))