diff options
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 |