diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-03-03 11:21:43 -0800 |
---|---|---|
committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-04-27 16:01:59 -0700 |
commit | 550ccc52de231621c0bf0c05ae2a398eec37ff51 (patch) | |
tree | d4a5bd8d24560135784208c0fe35615b1d043249 /verif/tosa_ref_run.py | |
parent | cf6224e6e8ba4fc2984de3e542538c38e27c9f57 (diff) | |
download | reference_model-550ccc52de231621c0bf0c05ae2a398eec37ff51.tar.gz |
Replace serialization/ and verif/ with MLPlatform's serialization_lib submodule
- Remove Usage and Format
- Run black on verif/*.py scripts
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ie81515891eb0039540f614894f4b6b0e0e78ba74
Diffstat (limited to 'verif/tosa_ref_run.py')
-rw-r--r-- | verif/tosa_ref_run.py | 32 |
1 files changed, 19 insertions, 13 deletions
diff --git a/verif/tosa_ref_run.py b/verif/tosa_ref_run.py index 2035147..26c64da 100644 --- a/verif/tosa_ref_run.py +++ b/verif/tosa_ref_run.py @@ -19,6 +19,7 @@ import shlex import subprocess from tosa_test_runner import TosaTestRunner, run_sh_command + class TosaRefRunner(TosaTestRunner): def __init__(self, args, runnerArgs, testDir): super().__init__(args, runnerArgs, testDir) @@ -28,28 +29,33 @@ class TosaRefRunner(TosaTestRunner): # Uses arguments from the argParser args, not the runnerArgs args = self.args - ref_cmd = [ args.ref_model_path, - '-Csubgraph_file={}'.format(self.testDesc['tosa_file']), - '-Csubgraph_dir={}'.format(self.testDir), - '-Cinput_dir={}'.format(self.testDir), - '-Coutput_dir={}'.format(self.testDir), - '-Coutput_tensor_prefix=ref-', # Naming agreement with TosaSerializer - ] + ref_cmd = [ + args.ref_model_path, + "-Csubgraph_file={}".format(self.testDesc["tosa_file"]), + "-Csubgraph_dir={}".format(self.testDir), + "-Cinput_dir={}".format(self.testDir), + "-Coutput_dir={}".format(self.testDir), + "-Coutput_tensor_prefix=ref-", # Naming agreement with TosaSerializer + ] # Build up input tensor_name/filename list inputTensors = [] - for i in range(len(self.testDesc['ifm_placeholder'])): - inputTensors.append('{}:{}'.format(self.testDesc['ifm_placeholder'][i], self.testDesc['ifm_file'][i])) + for i in range(len(self.testDesc["ifm_placeholder"])): + inputTensors.append( + "{}:{}".format( + self.testDesc["ifm_placeholder"][i], self.testDesc["ifm_file"][i] + ) + ) - ref_cmd.append('-Cinput_tensor={}'.format(','.join(inputTensors))) + ref_cmd.append("-Cinput_tensor={}".format(",".join(inputTensors))) if args.ref_debug: - ref_cmd.extend(['-dALL', '-l{}'.format(args.ref_debug)]) + ref_cmd.extend(["-dALL", "-l{}".format(args.ref_debug)]) if args.ref_intermediates: - ref_cmd.extend(['-Ddump_intermediates=1']) + ref_cmd.extend(["-Ddump_intermediates=1"]) - expectedFailure = self.testDesc['expected_failure'] + expectedFailure = self.testDesc["expected_failure"] try: run_sh_command(self.args, ref_cmd) |