diff options
| author | James Ward <james.ward@arm.com> | 2022-10-19 12:20:31 +0100 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-11-09 12:19:51 +0000 |
| commit | 24dbc420aae556649f50e645bd94489dab2cc75a (patch) | |
| tree | 490345da43e9c5bae0f450ba05ffe85874077e0a /verif/generator/tosa_verif_build_tests.py | |
| parent | 3b0544c1e7463295c49a48a162ebb9a546326829 (diff) | |
| download | reference_model-24dbc420aae556649f50e645bd94489dab2cc75a.tar.gz | |
Add BF16 support to reference model
* Upgrade Eigen to 3.4.0 (for bfloat16 support) and add work-
arounds for reduce.any() and reduce.all() bugs (introduced
between 3.3.7 and 3.4.0)
* Truncation to bfloat16 now performed in eval() methods
Signed-off-by: James Ward <james.ward@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: If5f5c988d76d3d30790acf3b97081726b89205fe
Diffstat (limited to 'verif/generator/tosa_verif_build_tests.py')
| -rw-r--r-- | verif/generator/tosa_verif_build_tests.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py index 2fafacb..ab78b1a 100644 --- a/verif/generator/tosa_verif_build_tests.py +++ b/verif/generator/tosa_verif_build_tests.py @@ -5,6 +5,7 @@ import re from generator.tosa_test_gen import TosaTestGen from serializer.tosa_serializer import dtype_str_to_val +from serializer.tosa_serializer import DTypeNames # Used for parsing a comma-separated list of integers in a string @@ -150,13 +151,14 @@ def parseArgs(argv): help="Create tests with a particular input tensor rank", ) + # Used for parsing a comma-separated list of integers in a string parser.add_argument( "--target-dtype", dest="target_dtypes", action="append", default=None, type=lambda x: dtype_str_to_val(x), - help="Create test with a particular DType (may be repeated)", + help=f"Create test with a particular DType: [{', '.join([d.lower() for d in DTypeNames[1:]])}] (may be repeated)", ) parser.add_argument( |