diff options
Diffstat (limited to 'verif/generator/tosa_utils.py')
| -rw-r--r-- | verif/generator/tosa_utils.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 6558bf8..cfe7cc6 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -64,6 +64,11 @@ def dtypeWidth(dtype): raise Exception(f"Unknown dtype, cannot determine width: {dtype}") +def dtypeIsFloat(dtype): + """Is floating point data type""" + return dtype in (DType.BF16, DType.FP16, DType.FP32, DType.FP8E4M3, DType.FP8E5M2) + + def dtypeIsSupportedByCompliance(dtype): """Types supported by the new data generation and compliance flow.""" if isinstance(dtype, list) or isinstance(dtype, tuple): |