diff options
Diffstat (limited to 'verif/generator/datagenerator.py')
-rw-r--r-- | verif/generator/datagenerator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/generator/datagenerator.py b/verif/generator/datagenerator.py index b5ef35d..743475c 100644 --- a/verif/generator/datagenerator.py +++ b/verif/generator/datagenerator.py @@ -78,7 +78,7 @@ class GenerateLibrary: size_bytes = size * 2 # Create buffer of bytes and initialize to zero buffer = (ct.c_ubyte * size_bytes)(0) - elif dtype == "INT32": + elif dtype == "INT32" or dtype == "SHAPE": # Create buffer and initialize to zero buffer = (ct.c_int32 * size)(0) size_bytes = size * 4 |