diff options
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r-- | verif/generator/tosa_arg_gen.py | 33 |
1 files changed, 27 insertions, 6 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index fed91f6..05a7d2b 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -1445,19 +1445,40 @@ class TosaArgGen: if error_name == ErrorIf.WrongOutputType: dtypeList = TosaErrorIfArgGen.eiCastErrorIf(testGen, inDtype) elif inDtype == DType.INT8: - dtypeList = [DType.BOOL, DType.INT16, DType.INT32, DType.FP32] + dtypeList = [ + DType.BOOL, + DType.INT16, + DType.INT32, + DType.FP16, + DType.BF16, + DType.FP32, + ] elif inDtype == DType.INT16: - dtypeList = [DType.BOOL, DType.INT8, DType.INT32, DType.FP32] + dtypeList = [ + DType.BOOL, + DType.INT8, + DType.INT32, + DType.FP16, + DType.BF16, + DType.FP32, + ] elif inDtype == DType.INT32: - dtypeList = [DType.BOOL, DType.INT8, DType.INT16, DType.FP32] + dtypeList = [ + DType.BOOL, + DType.INT8, + DType.INT16, + DType.FP16, + DType.BF16, + DType.FP32, + ] elif inDtype == DType.BOOL: dtypeList = [DType.INT8, DType.INT16, DType.INT32] elif inDtype == DType.FP16: - dtypeList = [DType.INT8, DType.INT16, DType.INT32] + dtypeList = [DType.INT8, DType.INT16, DType.INT32, DType.FP32] elif inDtype == DType.BF16: - dtypeList = [DType.INT8, DType.INT16, DType.INT32] + dtypeList = [DType.INT8, DType.INT16, DType.INT32, DType.FP32] elif inDtype == DType.FP32: - dtypeList = [DType.INT8, DType.INT16, DType.INT32] + dtypeList = [DType.INT8, DType.INT16, DType.INT32, DType.FP16, DType.BF16] elif error_name == ErrorIf.WrongInputType: # Pick some potentially correct output type for incorrect input type dtypeList = [DType.BOOL, DType.INT8, DType.INT16, DType.FP32] |