aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/datagenerator.py
diff options
context:
space:
mode:
Diffstat (limited to 'verif/generator/datagenerator.py')
-rw-r--r--verif/generator/datagenerator.py2
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