diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-15 11:00:49 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-16 21:24:34 +0000 |
| commit | 30a41db0c89f8209ab710c1d312fd6697107a41b (patch) | |
| tree | 3ebff243aa29fa2f538064852c3fa2950c4f96a3 /verif/generator/tosa_test_gen.py | |
| parent | 9a758384d1066ade713311940f3d15c860f90866 (diff) | |
| download | reference_model-30a41db0c89f8209ab710c1d312fd6697107a41b.tar.gz | |
Fix FP16, BF16 data ranges for conformance tests
Enable use of data ranges for old data gen path as well
as the new generate library path, so that FP16 and BF16 test data
is produced within the correct ranges.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I749870a3112f8c3a75f4d16b8322c813fbf977cd
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
| -rw-r--r-- | verif/generator/tosa_test_gen.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 3e5aee8..9dc3199 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -176,8 +176,11 @@ class TosaTestGen: # Inclusive range: low <= range <= high return (rng[0], rng[1] - 1) - def getRandTensor(self, shape, dtype): - low, high = self.getDTypeRange(dtype) + def getRandTensor(self, shape, dtype, data_range=None): + if data_range is None: + low, high = self.getDTypeRange(dtype) + else: + low, high = data_range if dtype == DType.BOOL: return np.bool_(self.rng.choice(a=[False, True], size=shape)) |