aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_arg_gen.py
diff options
context:
space:
mode:
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r--verif/generator/tosa_arg_gen.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 791fbf7..0203513 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1426,7 +1426,7 @@ class TosaArgGen:
raise Exception("Unexpected input dtype: {}".format(inDtype))
for dtype in dtypeList:
- arg_list.append(("out{}".format(DTypeNames[dtype]), [dtype]))
+ arg_list.append(("out{}".format(testGen.typeStr(dtype)), [dtype]))
return arg_list
@@ -1518,7 +1518,7 @@ class TosaArgGen:
arg_list.append(
(
"out{}_sc{}_dr{}_pc{}".format(
- DTypeNames[outDtype],
+ testGen.typeStr(outDtype),
int(scale32),
int(double_round),
int(per_channel),