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 0dd60e5..b5ef35d 100644 --- a/verif/generator/datagenerator.py +++ b/verif/generator/datagenerator.py @@ -78,6 +78,10 @@ class GenerateLibrary: size_bytes = size * 2 # Create buffer of bytes and initialize to zero buffer = (ct.c_ubyte * size_bytes)(0) + elif dtype == "INT32": + # Create buffer and initialize to zero + buffer = (ct.c_int32 * size)(0) + size_bytes = size * 4 else: raise GenerateError(f"Unsupported data type {dtype}") |