diff options
Diffstat (limited to 'tests/nnapi/nnapi_test_generator/android-10/test_generator.py')
-rwxr-xr-x | tests/nnapi/nnapi_test_generator/android-10/test_generator.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/tests/nnapi/nnapi_test_generator/android-10/test_generator.py b/tests/nnapi/nnapi_test_generator/android-10/test_generator.py index 44bfd801d..b4b07cbda 100755 --- a/tests/nnapi/nnapi_test_generator/android-10/test_generator.py +++ b/tests/nnapi/nnapi_test_generator/android-10/test_generator.py @@ -989,7 +989,7 @@ class Example: )) else: assert False - # Fix for neurun: disable dynamic shape test generation + # Fix for onert: disable dynamic shape test generation #if Configuration.test_dynamic_output_shape: #self.variations = [[DefaultVariation(), DynamicOutputShapeConverter()]] #else: @@ -1026,7 +1026,18 @@ class Example: def AddVariations(self, *args, includeDefault=True, defaultName=None): self.variations.append([DefaultVariation(defaultName)] if includeDefault else []) - self.variations[-1].extend(ImplicitVariation.ImplicitConvertion(i) for i in args) + # NNFW Fix: remove float16 type variation test generation + variations = [] + for i in args: + variation = ImplicitVariation.ImplicitConvertion(i) + print(i, file=sys.stderr) + if not isinstance(i, ModelVariation) and type(i) is str: + if i == "float16": + continue + else: + variations.append(variation) + self.variations[-1].extend(variations) + #self.variations[-1].extend(ImplicitVariation.ImplicitConvertion(i) for i in args) return self def AddNchw(self, *args, includeDefault=True, defaultName="nhwc"): |