From b3e80cb1e386500d1747b929d16569b73239bc31 Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Thu, 12 Dec 2019 15:44:11 +0200
Subject: [PATCH] Bugfix - missing argument in post-quant training unitest
(thanks @barrh)
---
tests/test_post_train_quant.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_post_train_quant.py b/tests/test_post_train_quant.py
index d54ea33..200ea26 100644
--- a/tests/test_post_train_quant.py
+++ b/tests/test_post_train_quant.py
@@ -803,4 +803,4 @@ def test_linear_layer_wrapper_params_only(per_channel, bias):
x = torch.rand(5, in_features)
- _test_wts_only_quant(layer, x, per_channel, bias)
+ _test_wts_only_quant(layer, x, per_channel, bias, 8)
--
GitLab