This repository was archived by the owner on Mar 21, 2024. It is now read-only.
File tree 1 file changed +7
-4
lines changed 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change 3
3
All non-trivial proposed changes to the code base should be accompanied by tests.
4
4
5
5
Each PullRequest build will run all tests in the repository on CPU machines. One full test run is executed on a Windows
6
- agent, one on a Linux agent.
6
+ agent, one on a Linux agent. Tests with the pytest mark ` gpu ` or ` azureml ` will not be executed in this run.
7
7
8
8
In addition, ` pytest ` will be run as part of the small AzureML job (smoke test) that is part of the PR build.
9
9
In that test run, only specific tests will be executed. At present, it will be the tests that are marked
10
- with the ` pytest ` mark ` gpu ` . The AzureML job executes on a GPU VM, hence you can have tests for GPU-specific
11
- capabilities.
10
+ with the pytest mark ` gpu ` , ` cpu_and_gpu ` or ` azureml ` . The AzureML job executes on a GPU VM, hence you can have
11
+ tests for GPU-specific capabilities.
12
12
13
- To mark one of your tests for execution on the GPU, prefix the test as follows:
13
+ To mark one of your tests as requiring a GPU, prefix the test as follows:
14
14
15
15
@pytest.mark.gpu
16
16
def test_my_gpu_code() -> None:
17
17
...
18
18
19
+ Similarly, use the mark ` cpu_and_gpu ` for tests that should be run twice, once on the CPU and once on the GPU. For
20
+ tests that do not require a GPU, but have longer running times and are better suited to run on the PR build, use the
21
+ mark ` azureml ` .
You can’t perform that action at this time.
0 commit comments