diff options
| -rw-r--r-- | verif/conformance/tosa_base_profile_ops_info.json | 14 | ||||
| -rw-r--r-- | verif/generator/tosa_arg_gen.py | 4 |
2 files changed, 12 insertions, 6 deletions
diff --git a/verif/conformance/tosa_base_profile_ops_info.json b/verif/conformance/tosa_base_profile_ops_info.json index 9dd5592..ff88293 100644 --- a/verif/conformance/tosa_base_profile_ops_info.json +++ b/verif/conformance/tosa_base_profile_ops_info.json @@ -478,10 +478,10 @@ "shape": [], "type": [], "output_type": [ - "outINT8", - "outINT16", - "outINT32", - "outBOOL" + "outi8", + "outi16", + "outi32", + "outb" ] }, "permutes": [ @@ -2423,6 +2423,12 @@ ], [ "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16" + ], + [ + "--target-dtype", "int8", "--target-shape", "65533", 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), |