aboutsummaryrefslogtreecommitdiff
path: root/verif/frameworks/test_builder.py
diff options
context:
space:
mode:
authorTatWai Chong <tatwai.chong@arm.com>2023-02-22 18:38:01 -0800
committerTatWai Chong <tatwai.chong@arm.com>2023-02-24 14:54:16 -0800
commit2226f90d5a6c48a975045bc9e0419113ce764aaf (patch)
tree78903dabce38db04e938414919fd45142fc72f4f /verif/frameworks/test_builder.py
parent681e0a8085d19629eddfb3d76a2ef9c0a7110f3c (diff)
downloadreference_model-2226f90d5a6c48a975045bc9e0419113ce764aaf.tar.gz
Extend pad tests to include quantized type and explicit pad value.
Signed-off-by: TatWai Chong <tatwai.chong@arm.com> Change-Id: I4a49f45aa73044aff5b0a8b3dba58c1f52c1ae21
Diffstat (limited to 'verif/frameworks/test_builder.py')
-rw-r--r--verif/frameworks/test_builder.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index 407902a..d995a34 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -793,8 +793,9 @@ class TBuilder:
)
class Pad:
- def __init__(self, padding, name):
+ def __init__(self, padding, pad_const, name):
self.padding = padding
+ self.pad_const = pad_const
self.result_name = name
def eval(self, a):
@@ -802,7 +803,7 @@ class TBuilder:
a,
self.padding,
mode="CONSTANT",
- constant_values=0,
+ constant_values=self.pad_const,
name=self.result_name,
)