tt-torch uses pytest for all unit and model tests.
Tests are organized into unit tests for pytorch (tests/torch), unit tests for onnx (test/onns) and models (tests/models). They can be run locally by running:
source env/activate
pytest -svv tests/torch
Model tests (tests/models) have the option to run op-by-up, see overview. This allows for faster model bring-up as it allows users to find any potential issues in parallel. This is controlled by the --op_by_op
flag:
pytest -svv tests/models/albert --op_by_op