Modify the tests/scripts/unit_tests/config.ymlconfiguration file.
Configuration file structure
Framework name(megatron/flagscale/...):
set_environment: Commands to configure the environment.
coverage: Directories that require test coverage statistics (the actual path = framework name + directory name).
Subset name(models/transformer/...):
depth: Specify which levels of tests to run. Default is to run all (all), also can be a integer.
ignore: Specify folders, files, or functions to skip.
type: Batch test or independent test. Default is batch (batch), also can be (single).
Note: Default values do not need to be explicitly marked. If a folder uses all default configurations, it can be omitted from the configuration file.
Here is an example of adding your own testing folder.
flagscale:
subset:
your_folder:
type: batch
depth: all
ignore: pytest_file.py
Main features
pytest
command each time, check if the GPU is idle to avoid test failures caused by resource contention.How to run all unit tests locally
How to run specific unit tests locally
How to add unit tests
Without adding a new folder:
When adding a new folder:
tests/scripts/unit_tests/config.yml
configuration file.Configuration file structure
Framework name(megatron/flagscale/...):
Subset name(models/transformer/...):
all
), also can be a integer.batch
), also can be (single
).Note: Default values do not need to be explicitly marked. If a folder uses all default configurations, it can be omitted from the configuration file. Here is an example of adding your own testing folder.
How to run this example
In the future, we will add unit testing and functional testing for VLLM