Skip to content

Commit 7f331f2

Browse files
author
Yaman Umuroglu
committed
[Test] use larger vector for MultiThreshold sim benchmark
1 parent d405296 commit 7f331f2

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/custom_op/test_multithreshold.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -299,8 +299,8 @@ def test_multithreshold():
299299
def multithreshold_performance():
300300
# performance and random test
301301
np.random.seed(0)
302-
inputs = np.random.random((1, 256, 64, 64))
303-
thresholds = (np.array([[1, 2, 3, 4, 5, 6]]) - 0.5) / 6
302+
inputs = np.random.random((128, 1024, 1))
303+
thresholds = (np.array([list(range(254))]) - 0.5) / 6
304304
before = time.time()
305305
vec_results = multithreshold(inputs, thresholds)
306306
after = time.time()

0 commit comments

Comments
 (0)