From 550ccc52de231621c0bf0c05ae2a398eec37ff51 Mon Sep 17 00:00:00 2001 From: Kevin Cheng Date: Wed, 3 Mar 2021 11:21:43 -0800 Subject: 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 Change-Id: Ie81515891eb0039540f614894f4b6b0e0e78ba74 --- verif/tosa_ref_run.py | 32 +++++++++++++++++++------------- 1 file changed, 19 insertions(+), 13 deletions(-) (limited to 'verif/tosa_ref_run.py') 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) -- cgit v1.2.1