diff options
Diffstat (limited to 'verif/generator/datagenerator.py')
-rw-r--r-- | verif/generator/datagenerator.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/generator/datagenerator.py b/verif/generator/datagenerator.py index 743475c..c63a2d5 100644 --- a/verif/generator/datagenerator.py +++ b/verif/generator/datagenerator.py @@ -82,6 +82,10 @@ class GenerateLibrary: # Create buffer and initialize to zero buffer = (ct.c_int32 * size)(0) size_bytes = size * 4 + elif dtype == "INT8": + size_bytes = size + # Create buffer of bytes and initialize to zero + buffer = (ct.c_ubyte * size_bytes)(0) else: raise GenerateError(f"Unsupported data type {dtype}") |