Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 1 | """TOSA test runner module for the Reference Model.""" |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 2 | # Copyright (c) 2020-2023, ARM Limited. |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 3 | # SPDX-License-Identifier: Apache-2.0 |
| 4 | from enum import IntEnum |
| 5 | from enum import unique |
| 6 | |
| 7 | from runner.run_command import run_sh_command |
| 8 | from runner.run_command import RunShCommandError |
| 9 | from runner.tosa_test_runner import TosaTestRunner |
| 10 | |
| 11 | |
| 12 | @unique |
| 13 | class TosaRefReturnCode(IntEnum): |
| 14 | """Return codes from the Tosa Reference Model.""" |
| 15 | |
| 16 | VALID = 0 |
| 17 | UNPREDICTABLE = 1 |
| 18 | ERROR = 2 |
| 19 | |
| 20 | |
| 21 | class TosaSUTRunner(TosaTestRunner): |
| 22 | """TOSA Reference Model runner.""" |
| 23 | |
Jeremy Johnson | e4b08ff | 2022-09-15 10:38:17 +0100 | [diff] [blame] | 24 | def __init__(self, args, runnerArgs, testDirPath): |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 25 | """Initialize using the given test details.""" |
Jeremy Johnson | e4b08ff | 2022-09-15 10:38:17 +0100 | [diff] [blame] | 26 | super().__init__(args, runnerArgs, testDirPath) |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 27 | |
Jeremy Johnson | e2b5e87 | 2023-09-14 17:02:09 +0100 | [diff] [blame] | 28 | # Don't do any compliance runs |
| 29 | self.compliance = False |
| 30 | |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 31 | def runTestGraph(self): |
| 32 | """Run the test on the reference model.""" |
| 33 | # Build up the TOSA reference command line |
| 34 | # Uses arguments from the argParser args, not the runnerArgs |
| 35 | args = self.args |
| 36 | |
| 37 | # Call Reference model with description file to provide all file details |
| 38 | cmd = [ |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 39 | str(args.ref_model_path), |
| 40 | f"--tosa_level={args.tosa_level}", |
| 41 | f"--operator_fbs={str(args.schema_path)}", |
| 42 | f"--test_desc={self.descFile}", |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 43 | ] |
| 44 | |
| 45 | # Specific debug options for reference model |
| 46 | if args.ref_debug: |
Jeremy Johnson | 93d4390 | 2022-09-27 12:26:14 +0100 | [diff] [blame] | 47 | cmd.extend(["-d", "ALL", "-l", args.ref_debug]) |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 48 | |
| 49 | if args.ref_intermediates: |
Jeremy Johnson | 93d4390 | 2022-09-27 12:26:14 +0100 | [diff] [blame] | 50 | cmd.extend(["--dump_intermediates", str(args.ref_intermediates)]) |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 51 | |
Jeremy Johnson | e2b5e87 | 2023-09-14 17:02:09 +0100 | [diff] [blame] | 52 | if args.precise_mode or self.compliance: |
Tai Ly | a4d748b | 2023-03-28 22:06:56 +0000 | [diff] [blame] | 53 | cmd.extend(["--precise_mode=1"]) |
| 54 | |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 55 | # Run command and interpret tosa graph result via process return codes |
| 56 | graphMessage = None |
| 57 | try: |
| 58 | run_sh_command(cmd, self.args.verbose, capture_output=True) |
| 59 | graphResult = TosaTestRunner.TosaGraphResult.TOSA_VALID |
| 60 | except RunShCommandError as e: |
| 61 | graphMessage = e.stderr |
| 62 | if e.return_code == TosaRefReturnCode.ERROR: |
| 63 | graphResult = TosaTestRunner.TosaGraphResult.TOSA_ERROR |
| 64 | elif e.return_code == TosaRefReturnCode.UNPREDICTABLE: |
| 65 | graphResult = TosaTestRunner.TosaGraphResult.TOSA_UNPREDICTABLE |
| 66 | else: |
| 67 | graphResult = TosaTestRunner.TosaGraphResult.OTHER_ERROR |
Jeremy Johnson | 015c355 | 2022-02-23 12:15:03 +0000 | [diff] [blame] | 68 | if not self.args.verbose: |
| 69 | print(e) |
Jeremy Johnson | be1a940 | 2021-12-15 17:14:56 +0000 | [diff] [blame] | 70 | except Exception as e: |
| 71 | print(e) |
| 72 | graphMessage = str(e) |
| 73 | graphResult = TosaTestRunner.TosaGraphResult.OTHER_ERROR |
| 74 | |
| 75 | # Return graph result and message |
| 76 | return graphResult, graphMessage |