unit-test.yml 415 B

12345678910111213141516171819202122
  1. # Copyright (c) Microsoft Corporation.
  2. # Licensed under the MIT license.
  3. trigger:
  4. - main
  5. pool:
  6. name: Tutel CI
  7. vmImage: ubuntu-latest
  8. container:
  9. image: nvcr.io/nvidia/pytorch:21.08-py3
  10. options: '--gpus=all'
  11. steps:
  12. - script: |
  13. python3 -m pip install .[test]
  14. displayName: Install dependencies
  15. - script: |
  16. python3 setup.py test
  17. displayName: Run unit tests
  18. timeoutInMinutes: 30