diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-28 15:53:21 +0000 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-04-11 15:02:57 +0100 |
| commit | 18a379d99ad10002b3cf6eda086457179221cc22 (patch) | |
| tree | 9b90a31f846035236cbecb9cde379dee66b6f0c3 /verif/generator/datagenerator.py | |
| parent | 3f3de01fa87246161e47c15fd6c44f710b86f3e7 (diff) | |
| download | reference_model-18a379d99ad10002b3cf6eda086457179221cc22.tar.gz | |
Add rank 0 testing support
Default test range is now rank 0 to 3 instead of 1 to 4
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ibde66b60b58de9f4a3852a3807c01f8dae61206f
Diffstat (limited to 'verif/generator/datagenerator.py')
| -rw-r--r-- | verif/generator/datagenerator.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/verif/generator/datagenerator.py b/verif/generator/datagenerator.py index c63a2d5..9f3d29d 100644 --- a/verif/generator/datagenerator.py +++ b/verif/generator/datagenerator.py @@ -68,7 +68,11 @@ class GenerateLibrary: def _create_buffer(self, dtype: str, shape: tuple): """Helper to create a buffer of the required type.""" - size = np.prod(shape) + if shape: + size = np.prod(shape) + else: + # Rank 0 + size = 1 if dtype == "FP32": # Create buffer and initialize to zero |