From cccd7557c8110c4fd2a5a781b34b3846e762f8b5 Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Wed, 11 Jan 2023 21:14:25 +0300 Subject: [PATCH] update input data for test of dense --- tests/python/unittest/test_meta_schedule_16x4_integration.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/unittest/test_meta_schedule_16x4_integration.py b/tests/python/unittest/test_meta_schedule_16x4_integration.py index 0d9a789f5377d..4498e848c0747 100644 --- a/tests/python/unittest/test_meta_schedule_16x4_integration.py +++ b/tests/python/unittest/test_meta_schedule_16x4_integration.py @@ -112,8 +112,8 @@ def _relay_dense(m, n, k): relay_mod = tvm.IRModule.from_expr(out) data = np.random.uniform(1, 10, size=(m, k)).astype("uint8") params = { - "weight": np.random.uniform(1, 10, size=(n, k)).astype("int8"), - "bias": np.random.uniform(1, 10, size=(n,)).astype("int32"), + "weight": np.random.uniform(-10, 10, size=(n, k)).astype("int8"), + "bias": np.random.uniform(-10, 10, size=(n,)).astype("int32"), } def f_check(lib, dev):