diff options
author | James Ward <james.ward@arm.com> | 2022-12-07 15:48:47 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-01-19 16:31:59 +0000 |
commit | 3407125ce2a470216bd54f8bb3ab5216f617c1be (patch) | |
tree | 7e365e7c25618f87cb869fd472ef3c6903f3352a /verif/generator/tosa_test_gen.py | |
parent | 657af86089aa60e12e59943146247c8c706be573 (diff) | |
download | reference_model-3407125ce2a470216bd54f8bb3ab5216f617c1be.tar.gz |
Create MI tests for Activation: CLAMP; Data Layout: PAD
* Existing float attributes now serialized as bytes
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I415276706b9daf0893e3a59189f387f872ff07c2
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index d799eb0..c29763b 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -1072,10 +1072,15 @@ class TosaTestGen: return None attr = ts.TosaSerializerAttribute() - if a.dtype in (DType.FP16, DType.BF16, DType.FP32): - attr.ClampAttribute(0, 0, min_val, max_val) + if a.dtype in (DType.BF16, DType.FP16, DType.FP32): + if a.dtype == DType.FP16: + # Non-tensor fp16 ops take fp16 values as fp32 in reference_model + min_val = min_val.astype(np.float32) + max_val = max_val.astype(np.float32) + + attr.ClampAttribute(self.ser.builder, 0, 0, min_val, max_val) else: - attr.ClampAttribute(min_val, max_val, 0, 0) + attr.ClampAttribute(self.ser.builder, min_val, max_val, 0, 0) self.ser.addOperator(op["op"], input_list, output_list, attr) return result_tens @@ -1221,7 +1226,9 @@ class TosaTestGen: result_tens = OutputShaper.padOp(self.ser, self.rng, a, padding, error_name) attr = ts.TosaSerializerAttribute() - attr.PadAttribute(padding.flatten(), pad_const_int, pad_const_float) + attr.PadAttribute( + self.ser.builder, padding.flatten(), pad_const_int, pad_const_float + ) # Invalidate Input/Output list for error if checks. input_list = [a.name] |