config.yml.in 25.8 KB
Newer Older
1
2
3
4
version: 2.1

# How to test the Linux jobs:
#   - Install CircleCI local CLI: https://circleci.com/docs/2.0/local-cli/
5
6
#   - circleci config process .circleci/config.yml > gen.yml && circleci local execute -c gen.yml --job binary_linux_wheel_py3.8
#     - Replace binary_linux_wheel_py3.8 with the name of the job you want to test.
Edward Z. Yang's avatar
Edward Z. Yang committed
7
#       Job names are 'name:' key.
8

9
executors:
10
11
12
13
14
15
16
  windows-cpu:
    machine:
      resource_class: windows.xlarge
      image: windows-server-2019-vs2019:stable
      shell: bash.exe

  windows-gpu:
17
    machine:
18
19
      resource_class: windows.gpu.nvidia.medium
      image: windows-server-2019-nvidia:stable
20
21
      shell: bash.exe

22
23
24
25
26
commands:
  checkout_merge:
    description: "checkout merge branch"
    steps:
      - checkout
27
28
29
30
31
#     - run:
#         name: Checkout merge branch
#         command: |
#           set -ex
#           BRANCH=$(git rev-parse --abbrev-ref HEAD)
32
#           if [[ "$BRANCH" != "main" ]]; then
33
34
35
#             git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
#             git checkout "merged/$CIRCLE_BRANCH"
#           fi
36
37
38
39
40
41
42
43
44
45
46
47
  designate_upload_channel:
    description: "inserts the correct upload channel into ${BASH_ENV}"
    steps:
      - run:
          name: adding UPLOAD_CHANNEL to BASH_ENV
          command: |
            our_upload_channel=nightly
            # On tags upload to test instead
            if [[ -n "${CIRCLE_TAG}" ]]; then
              our_upload_channel=test
            fi
            echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
48

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
  brew_update:
    description: "Update Homebrew and install base formulae"
    steps:
      - run:
          name: Update Homebrew
          no_output_timeout: "10m"
          command: |
            set -ex

            # Update repositories manually.
            # Running `brew update` produces a comparison between the
            # current checkout and the updated checkout, which takes a
            # very long time because the existing checkout is 2y old.
            for path in $(find /usr/local/Homebrew -type d -name .git)
            do
            cd $path/..
            git fetch --depth=1 origin
            git reset --hard origin/master
            done

            export HOMEBREW_NO_AUTO_UPDATE=1

            # Install expect and moreutils so that we can call `unbuffer` and `ts`.
            # moreutils installs a `parallel` executable by default, which conflicts
            # with the executable from the GNU `parallel`, so we must unlink GNU
            # `parallel` first, and relink it afterwards.
            brew install coreutils
            brew unlink parallel
            brew install moreutils
            brew link parallel --overwrite
            brew install expect

  brew_install:
    description: "Install Homebrew formulae"
    parameters:
      formulae:
        type: string
        default: ""
    steps:
      - run:
          name: Install << parameters.formulae >>
          no_output_timeout: "10m"
          command: |
            set -ex
            export HOMEBREW_NO_AUTO_UPDATE=1
            brew install << parameters.formulae >>

  run_brew_for_ios_build:
    steps:
      - brew_update
      - brew_install:
          formulae: libtool

Philip Meier's avatar
Philip Meier committed
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
  apt_install:
    parameters:
      args:
        type: string
      descr:
        type: string
        default: ""
      update:
        type: boolean
        default: true
    steps:
      - run:
          name: >
            <<^ parameters.descr >> apt install << parameters.args >> <</ parameters.descr >>
            <<# parameters.descr >> << parameters.descr >>            <</ parameters.descr >>
          command: |
            <<# parameters.update >> sudo apt update -qy  <</ parameters.update >>
            sudo apt install << parameters.args >>

  pip_install:
    parameters:
      args:
        type: string
      descr:
        type: string
        default: ""
      user:
        type: boolean
        default: true
    steps:
      - run:
          name: >
            <<^ parameters.descr >> pip install << parameters.args >> <</ parameters.descr >>
            <<# parameters.descr >> << parameters.descr >>            <</ parameters.descr >>
          command: >
            pip install
            <<# parameters.user >> --user <</ parameters.user >>
            --progress-bar=off
            << parameters.args >>

  install_torchvision:
    parameters:
      editable:
        type: boolean
        default: true
    steps:
      - pip_install:
149
          args: --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu
Philip Meier's avatar
Philip Meier committed
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
          descr: Install PyTorch from nightly releases
      - pip_install:
          args: --no-build-isolation <<# parameters.editable >> --editable <</ parameters.editable >> .
          descr: Install torchvision <<# parameters.editable >> in editable mode <</ parameters.editable >>

  # Most of the test suite is handled by the `unittest` jobs, with completely different workflow and setup.
  # This command can be used if only a selection of tests need to be run, for ad-hoc files.
  run_tests_selective:
    parameters:
      file_or_dir:
        type: string
    steps:
      - run:
          name: Install test utilities
          command: pip install --progress-bar=off pytest pytest-mock
      - run:
          name: Run tests
          command: pytest --junitxml=test-results/junit.xml -v --durations 20 <<parameters.file_or_dir>>
      - store_test_results:
          path: test-results

171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
  download_model_weights:
    parameters:
      extract_roots:
        type: string
        default: "torchvision/models"
      background:
        type: boolean
        default: true
    steps:
      - apt_install:
          args: parallel wget
          descr: Install download utilitites
      - run:
          name: Download model weights
          background: << parameters.background >>
          command: |
            mkdir -p ~/.cache/torch/hub/checkpoints
            python scripts/collect_model_urls.py << parameters.extract_roots >> \
                | parallel -j0 'wget --no-verbose -O ~/.cache/torch/hub/checkpoints/`basename {}` {}\?source=ci'

191
192
binary_common: &binary_common
  parameters:
193
    # Edit these defaults to do a release
194
195
196
197
198
    build_version:
      description: "version number of release binary; by default, build a nightly"
      type: string
      default: ""
    pytorch_version:
199
      description: "PyTorch version to build against; by default, use a nightly"
200
      type: string
201
      default: ""
202
203
    # Don't edit these
    python_version:
204
      description: "Python version to build against (e.g., 3.8)"
205
      type: string
Edward Z. Yang's avatar
Edward Z. Yang committed
206
207
    cu_version:
      description: "CUDA version to build against, in CU format (e.g., cpu or cu100)"
208
      type: string
guyang3532's avatar
guyang3532 committed
209
      default: "cpu"
210
211
212
213
    unicode_abi:
      description: "Python 2.7 wheel only: whether or not we are cp27mu (default: no)"
      type: string
      default: ""
Edward Z. Yang's avatar
Edward Z. Yang committed
214
215
216
    wheel_docker_image:
      description: "Wheel only: what docker image to use"
      type: string
217
      default: ""
218
219
220
221
    conda_docker_image:
      description: "Conda only: what docker image to use"
      type: string
      default: "pytorch/conda-builder:cpu"
222
223
224
225
  environment:
    PYTHON_VERSION: << parameters.python_version >>
    PYTORCH_VERSION: << parameters.pytorch_version >>
    UNICODE_ABI: << parameters.unicode_abi >>
Edward Z. Yang's avatar
Edward Z. Yang committed
226
    CU_VERSION: << parameters.cu_version >>
227
    MACOSX_DEPLOYMENT_TARGET: 10.9
228

229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
torchvision_ios_params: &torchvision_ios_params
  parameters:
    build_environment:
      type: string
      default: ""
    ios_arch:
      type: string
      default: ""
    ios_platform:
      type: string
      default: ""
  environment:
    BUILD_ENVIRONMENT: << parameters.build_environment >>
    IOS_ARCH: << parameters.ios_arch >>
    IOS_PLATFORM: << parameters.ios_platform >>

245
246
247
248
249
250
251
252
torchvision_android_params: &torchvision_android_params
  parameters:
    build_environment:
      type: string
      default: ""
  environment:
    BUILD_ENVIRONMENT: << parameters.build_environment >>

guyang3532's avatar
guyang3532 committed
253
254
255
256
257
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
    - image: torchvision/smoke_test:latest

258
259
260
jobs:
  circleci_consistency:
    docker:
261
      - image: cimg/python:3.8
262
263
    steps:
      - checkout
Philip Meier's avatar
Philip Meier committed
264
265
      - pip_install:
          args: jinja2 pyyaml
266
      - run:
Philip Meier's avatar
Philip Meier committed
267
          name: Check CircleCI config consistency
268
269
270
271
272
273
274
          command: |
            python .circleci/regenerate.py
            git diff --exit-code || (echo ".circleci/config.yml not in sync with config.yml.in! Run .circleci/regenerate.py to update config"; exit 1)

  binary_linux_wheel:
    <<: *binary_common
    docker:
Edward Z. Yang's avatar
Edward Z. Yang committed
275
      - image: << parameters.wheel_docker_image >>
276
277
    resource_class: 2xlarge+
    steps:
278
      - checkout_merge
279
      - designate_upload_channel
280
281
282
283
284
285
      - run:
          name: Build conda packages
          no_output_timeout: 30m
          command: |
              set -ex
              packaging/build_wheel.sh
286
287
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
288
289
290
291
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
292
293
294
295

  binary_linux_conda:
    <<: *binary_common
    docker:
296
      - image: "<< parameters.conda_docker_image >>"
297
298
    resource_class: 2xlarge+
    steps:
299
      - checkout_merge
300
      - designate_upload_channel
301
302
303
304
305
306
      - run:
          name: Build conda packages
          no_output_timeout: 30m
          command: |
              set -ex
              packaging/build_conda.sh
307
308
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
309
310
311
312
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"
313
314
      - store_test_results:
          path: build_results/
315

Francisco Massa's avatar
Francisco Massa committed
316
  binary_win_conda:
317
318
    <<: *binary_common
    executor: windows-cpu
319
320
    steps:
      - checkout_merge
321
      - designate_upload_channel
322
323
      - run:
          name: Build conda packages
324
          no_output_timeout: 30m
325
          command: |
326
327
328
329
330
331
332
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
            packaging/build_conda.sh
333
            rm /C/tools/miniconda3/conda-bld/win-64/vs${VC_YEAR}*.tar.bz2
334
      - store_artifacts:
335
          path: C:/tools/miniconda3/conda-bld/win-64
336
      - persist_to_workspace:
337
          root: C:/tools/miniconda3/conda-bld/win-64
338
339
340
341
342
          paths:
            - "*"
      - store_test_results:
          path: build_results/

343
  binary_win_wheel:
344
345
    <<: *binary_common
    executor: windows-cpu
346
347
    steps:
      - checkout_merge
348
      - designate_upload_channel
349
350
      - run:
          name: Build wheel packages
351
          no_output_timeout: 30m
352
          command: |
353
354
355
356
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_wheel.sh
357
      - store_artifacts:
358
          path: dist
359
      - persist_to_workspace:
360
          root: dist
361
362
363
364
365
          paths:
            - "*"
      - store_test_results:
          path: build_results/

366
367
368
  binary_macos_wheel:
    <<: *binary_common
    macos:
369
      xcode: "14.0"
370
    steps:
371
      - checkout_merge
372
      - designate_upload_channel
373
374
375
376
377
378
379
380
381
382
383
      - run:
          # Cannot easily deduplicate this as source'ing activate
          # will set environment variables which we need to propagate
          # to build_wheel.sh
          command: |
            curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
            sh conda.sh -b
            source $HOME/miniconda3/bin/activate
            packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
384
385
386
387
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
388

389
390
391
  binary_ios_build:
    <<: *torchvision_ios_params
    macos:
392
      xcode: "14.0"
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
    steps:
    - attach_workspace:
        at: ~/workspace
    - checkout
    - run_brew_for_ios_build
    - run:
        name: Build
        no_output_timeout: "1h"
        command: |
          script="/Users/distiller/project/.circleci/unittest/ios/scripts/binary_ios_build.sh"
          cat "$script"
          source "$script"
    - persist_to_workspace:
        root: /Users/distiller/workspace/
        paths: ios

  binary_ios_upload:
    <<: *torchvision_ios_params
    macos:
412
      xcode: "14.0"
413
414
415
416
417
418
419
420
421
422
423
424
425
    steps:
    - attach_workspace:
        at: ~/workspace
    - checkout
    - run_brew_for_ios_build
    - run:
        name: Upload
        no_output_timeout: "1h"
        command: |
          script="/Users/distiller/project/.circleci/unittest/ios/scripts/binary_ios_upload.sh"
          cat "$script"
          source "$script"

426
427
428
  binary_android_build:
    <<: *torchvision_android_params
    docker:
429
      - image: cimg/android:2021.08-ndk
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
    resource_class: xlarge
    steps:
    - attach_workspace:
        at: ~/workspace
    - checkout
    - run:
        name: Build
        no_output_timeout: "1h"
        command: |
          script="/home/circleci/project/.circleci/unittest/android/scripts/binary_android_build.sh"
          cat "$script"
          source "$script"
    - store_artifacts:
        path: ~/workspace/artifacts

  binary_android_upload:
    <<: *torchvision_android_params
    docker:
448
      - image: cimg/android:2021.08-ndk
449
450
451
452
453
454
455
456
457
458
459
460
461
    resource_class: xlarge
    steps:
    - attach_workspace:
        at: ~/workspace
    - checkout
    - run:
        name: Upload
        no_output_timeout: "1h"
        command: |
          script="/home/circleci/project/.circleci/unittest/android/scripts/binary_android_upload.sh"
          cat "$script"
          source "$script"

462
463
464
  binary_macos_conda:
    <<: *binary_common
    macos:
465
      xcode: "14.0"
466
    steps:
467
      - checkout_merge
468
      - designate_upload_channel
469
470
471
472
473
474
475
476
477
      - run:
          command: |
            curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
            sh conda.sh -b
            source $HOME/miniconda3/bin/activate
            conda install -yq conda-build
            packaging/build_conda.sh
      - store_artifacts:
          path: /Users/distiller/miniconda3/conda-bld/osx-64
Edward Z. Yang's avatar
Edward Z. Yang committed
478
479
480
481
      - persist_to_workspace:
          root: /Users/distiller/miniconda3/conda-bld/osx-64
          paths:
            - "*"
482
483
      - store_test_results:
          path: build_results/
484

Edward Z. Yang's avatar
Edward Z. Yang committed
485
486
487
488
489
490
491
  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
492
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
493
494
495
496
497
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
498
            anaconda  -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/*.tar.bz2 -u "pytorch-${UPLOAD_CHANNEL}" --label main --no-progress --force
Edward Z. Yang's avatar
Edward Z. Yang committed
499
500
501

  # Requires org-member context
  binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
502
503
504
505
    parameters:
      subfolder:
        description: "What whl subfolder to upload to, e.g., blank or cu100/ (trailing slash is important)"
        type: string
Edward Z. Yang's avatar
Edward Z. Yang committed
506
    docker:
507
      - image: cimg/python:3.8
Edward Z. Yang's avatar
Edward Z. Yang committed
508
509
510
    steps:
      - attach_workspace:
          at: ~/workspace
511
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
512
      - checkout
Philip Meier's avatar
Philip Meier committed
513
514
      - pip_install:
          args: awscli
Edward Z. Yang's avatar
Edward Z. Yang committed
515
516
517
518
519
520
521
522
      - run:
          command: |
            export PATH="$HOME/.local/bin:$PATH"
            # Prevent credential from leaking
            set +x
            export AWS_ACCESS_KEY_ID="${PYTORCH_BINARY_AWS_ACCESS_KEY_ID}"
            export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}"
            set -x
Edward Z. Yang's avatar
Edward Z. Yang committed
523
            for pkg in ~/workspace/*.whl; do
524
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
525
            done
Edward Z. Yang's avatar
Edward Z. Yang committed
526

guyang3532's avatar
guyang3532 committed
527
528
529
530
531
  smoke_test_linux_conda:
    <<: *smoke_test_common
    steps:
      - attach_workspace:
          at: ~/workspace
532
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
      - run:
          name: install binaries
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
            conda install -v -y -c pytorch-nightly pytorch
            conda install -v -y $(ls ~/workspace/torchvision*.tar.bz2)
      - run:
          name: smoke test
          command: |
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
            python -c "import torchvision"

  smoke_test_linux_pip:
    <<: *smoke_test_common
    steps:
      - attach_workspace:
          at: ~/workspace
551
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
552
553
554
555
556
      - run:
          name: install binaries
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
Philip Meier's avatar
Philip Meier committed
557
558
      - pip_install:
          args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
guyang3532's avatar
guyang3532 committed
559
560
561
562
563
564
565
566
      - run:
          name: smoke test
          command: |
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
            python -c "import torchvision"

  smoke_test_docker_image_build:
    machine:
567
      image: ubuntu-2004:202104-01
guyang3532's avatar
guyang3532 committed
568
569
570
571
572
    resource_class: large
    environment:
      image_name: torchvision/smoke_test
    steps:
      - checkout
573
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
      - run:
          name: Build and push Docker image
          no_output_timeout: "1h"
          command: |
            set +x
            echo "${DOCKER_HUB_TOKEN}" | docker login --username "${DOCKER_HUB_USERNAME}" --password-stdin
            set -x
            cd .circleci/smoke_test/docker && docker build . -t ${image_name}:${CIRCLE_WORKFLOW_ID}
            docker tag ${image_name}:${CIRCLE_WORKFLOW_ID} ${image_name}:latest
            docker push ${image_name}:${CIRCLE_WORKFLOW_ID}
            docker push ${image_name}:latest

  smoke_test_win_conda:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - attach_workspace:
          at: ~/workspace
593
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
594
595
596
597
598
599
      - run:
          name: install binaries
          command: |
            set -x
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda env remove -n python${PYTHON_VERSION} || true
600
            conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
guyang3532's avatar
guyang3532 committed
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
            conda activate python${PYTHON_VERSION}
            conda install -v -y -c pytorch-nightly pytorch
            conda install -v -y $(ls ~/workspace/torchvision*.tar.bz2)
      - run:
          name: smoke test
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate python${PYTHON_VERSION}
            python -c "import torchvision"

  smoke_test_win_pip:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - attach_workspace:
          at: ~/workspace
618
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
619
620
621
622
623
624
625
      - run:
          name: install binaries
          command: |
            set -x
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
            conda activate python${PYTHON_VERSION}
Philip Meier's avatar
Philip Meier committed
626
627
      - pip_install:
          args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
guyang3532's avatar
guyang3532 committed
628
629
630
631
632
633
634
      - run:
          name: smoke test
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate python${PYTHON_VERSION}
            python -c "import torchvision"

635
636
637
638
639
640
  unittest_windows_cpu:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout
641
      - designate_upload_channel
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
      - restore_cache:
          {% raw %}
          keys:
            - env-v2-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
          {% endraw %}
      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - save_cache:
          {% raw %}
          key: env-v2-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
          {% endraw %}
          paths:
            - conda
            - env
      - run:
          name: Install torchvision
          command: .circleci/unittest/windows/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/windows/scripts/run_test.sh
      - run:
          name: Post process
          command: .circleci/unittest/windows/scripts/post_process.sh
      - store_test_results:
          path: test-results

  unittest_windows_gpu:
    <<: *binary_common
    executor:
      name: windows-gpu
    environment:
678
      CUDA_VERSION: "11.7"
679
      PYTHON_VERSION: << parameters.python_version >>
680
681
    steps:
      - checkout
682
      - designate_upload_channel
683
684
685
686
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
687
688
689
690
      - restore_cache:
          {% raw %}
          keys:
            - env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
691
          {% endraw %}
692
693
694
695
696
697
698
699
700
701
      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - save_cache:
          {% raw %}
          key: env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
          {% endraw %}
          paths:
            - conda
            - env
702
703
704
705
706
707
      - run:
          name: Install CUDA
          command: packaging/windows/internal/cuda_install.bat
      - run:
          name: Update CUDA driver
          command: packaging/windows/internal/driver_update.bat
708
709
710
711
712
713
714
715
716
717
718
      - run:
          name: Install torchvision
          command: .circleci/unittest/windows/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/windows/scripts/run_test.sh
      - run:
          name: Post process
          command: .circleci/unittest/windows/scripts/post_process.sh
      - store_test_results:
          path: test-results
Edward Z. Yang's avatar
Edward Z. Yang committed
719

720
721
722
  cmake_linux_cpu:
    <<: *binary_common
    docker:
723
      - image: "pytorch/manylinux-cpu"
724
725
726
    resource_class: 2xlarge+
    steps:
      - checkout_merge
727
      - designate_upload_channel
728
729
730
731
732
733
734
735
      - run:
          name: Setup conda
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run: packaging/build_cmake.sh

  cmake_linux_gpu:
    <<: *binary_common
    machine:
736
      image: ubuntu-2004-cuda-11.4:202110-01
737
    resource_class: gpu.nvidia.small
738
739
    steps:
      - checkout_merge
740
      - designate_upload_channel
741
742
743
744
745
      - run:
          name: Setup conda
          command: docker run -e CU_VERSION -e PYTHON_VERSION -e UNICODE_ABI -e PYTORCH_VERSION -t --gpus all -v $PWD:$PWD -w $PWD << parameters.wheel_docker_image >> .circleci/unittest/linux/scripts/setup_env.sh
      - run:
          name: Build torchvision C++ distribution and test
Andrey Talman's avatar
Andrey Talman committed
746
          no_output_timeout: 30m
747
          command: docker run -e CU_VERSION -e PYTHON_VERSION -e UNICODE_ABI -e PYTORCH_VERSION -e UPLOAD_CHANNEL -t --gpus all -v $PWD:$PWD -w $PWD << parameters.wheel_docker_image >> packaging/build_cmake.sh
748
749
750
751

  cmake_macos_cpu:
    <<: *binary_common
    macos:
752
      xcode: "14.0"
753
754
    steps:
      - checkout_merge
755
      - designate_upload_channel
756
757
758
759
760
      - run:
          command: |
            curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
            sh conda.sh -b
            source $HOME/miniconda3/bin/activate
761
            conda install -yq conda-build cmake python=<< parameters.python_version >>
762
763
764
765
766
767
768
769
            packaging/build_cmake.sh

  cmake_windows_cpu:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout_merge
770
      - designate_upload_channel
771
772
773
774
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
775
776
777
778
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            conda create -yn python39 python=3.9
            conda activate python39
779
780
781
782
783
784
785
786
            packaging/build_cmake.sh

  cmake_windows_gpu:
    <<: *binary_common
    executor:
      name: windows-gpu
    steps:
      - checkout_merge
787
      - designate_upload_channel
Andrey Talman's avatar
Andrey Talman committed
788
789
790
      - run:
          name: Update CUDA driver
          command: packaging/windows/internal/driver_update.bat
791
792
793
794
795
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
796
797
798
799
800
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate
            conda update -y conda
            conda create -yn python39 python=3.9
            conda activate python39
801
802
            packaging/build_cmake.sh

803

Edward Z. Yang's avatar
Edward Z. Yang committed
804
workflows:
Philip Meier's avatar
Philip Meier committed
805
  lint:
Edward Z. Yang's avatar
Edward Z. Yang committed
806
807
    jobs:
      - circleci_consistency
Philip Meier's avatar
Philip Meier committed
808
809
810

  build:
    jobs:
811
      {{ build_workflows(windows_latest_only=True) }}
812
      {{ ios_workflows() }}
813
      {{ android_workflows() }}
Edward Z. Yang's avatar
Edward Z. Yang committed
814

815
816
817
  unittest:
    jobs:
      {{ unittest_workflows() }}
818
819
820
821
822

  cmake:
    jobs:
      {{ cmake_workflows() }}

Edward Z. Yang's avatar
Edward Z. Yang committed
823
824
  nightly:
    jobs:
825
      {{ ios_workflows(nightly=True) }}
826
      {{ android_workflows(nightly=True) }}
827
      {{ build_workflows(prefix="nightly_", filter_branch="nightly", upload=True) }}
guyang3532's avatar
guyang3532 committed
828
829
830
831
832
833
834
  docker_build:
    triggers:
      - schedule:
          cron: "0 10 * * 0"
          filters:
            branches:
              only:
835
                - main
guyang3532's avatar
guyang3532 committed
836
837
838
    jobs:
      - smoke_test_docker_image_build:
          context: org-member