diff --git a/tank/tflitehub/lit.cfg.py b/tank/tflitehub/lit.cfg.py deleted file mode 100644 index af42ebec..00000000 --- a/tank/tflitehub/lit.cfg.py +++ /dev/null @@ -1,46 +0,0 @@ -import os -import sys - -import lit.formats -import lit.util - -import lit.llvm - -# Configuration file for the 'lit' test runner. -lit.llvm.initialize(lit_config, config) - -# name: The name of this test suite. -config.name = 'TFLITEHUB' - -config.test_format = lit.formats.ShTest() - -# suffixes: A list of file extensions to treat as test files. -config.suffixes = ['.py'] - -# test_source_root: The root path where tests are located. -config.test_source_root = os.path.dirname(__file__) - -#config.use_default_substitutions() -config.excludes = [ - 'coco_test_data.py', - 'imagenet_test_data.py', - 'lit.cfg.py', - 'lit.site.cfg.py', - 'manual_test.py', - 'squad_test_data.py', - 'test_util.py', -] - -config.substitutions.extend([ - ('%PYTHON', sys.executable), -]) - -config.environment['PYTHONPATH'] = ":".join(sys.path) - -project_root = os.path.dirname(os.path.dirname(__file__)) - -# Enable features based on -D FEATURES=hugetest,vulkan -# syntax. -features_param = lit_config.params.get('FEATURES') -if features_param: - config.available_features.update(features_param.split(',')) diff --git a/tank/tflitehub/ssd_mobilenet_v2_int8_test.py b/tank/tflitehub/ssd_mobilenet_v2_int8_test.py index c429de1d..5f901565 100644 --- a/tank/tflitehub/ssd_mobilenet_v2_int8_test.py +++ b/tank/tflitehub/ssd_mobilenet_v2_int8_test.py @@ -13,7 +13,7 @@ class SsdMobilenetV2Test(test_util.TFLiteModelTest): super(SsdMobilenetV2Test, self).__init__(model_path, *args, **kwargs) def compare_results(self, iree_results, tflite_results, details): - super(SsdMobilenetV1Test, self).compare_results(iree_results, tflite_results, details) + super(SsdMobilenetV2Test, self).compare_results(iree_results, tflite_results, details) for i in range(len(iree_results)): # Dequantize outputs. zero_point = details[i]['quantization_parameters']['zero_points'][0]