config.yml 124 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/
Edward Z. Yang's avatar
Edward Z. Yang committed
5
6
7
#   - circleci config process .circleci/config.yml > gen.yml && circleci local execute -c gen.yml --job binary_linux_wheel_py3.7
#     - Replace binary_linux_wheel_py3.7 with the name of the job you want to test.
#       Job names are 'name:' key.
8

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

  windows-gpu:
    machine:
      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

102
103
binary_common: &binary_common
  parameters:
104
    # Edit these defaults to do a release
105
106
107
108
109
    build_version:
      description: "version number of release binary; by default, build a nightly"
      type: string
      default: ""
    pytorch_version:
110
      description: "PyTorch version to build against; by default, use a nightly"
111
      type: string
112
      default: ""
113
114
115
116
    # Don't edit these
    python_version:
      description: "Python version to build against (e.g., 3.7)"
      type: string
Edward Z. Yang's avatar
Edward Z. Yang committed
117
118
    cu_version:
      description: "CUDA version to build against, in CU format (e.g., cpu or cu100)"
119
      type: string
guyang3532's avatar
guyang3532 committed
120
      default: "cpu"
121
122
123
124
    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
125
126
127
    wheel_docker_image:
      description: "Wheel only: what docker image to use"
      type: string
128
      default: "pytorch/manylinux-cuda102"
129
130
131
132
    conda_docker_image:
      description: "Conda only: what docker image to use"
      type: string
      default: "pytorch/conda-builder:cpu"
133
134
135
136
  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
137
    CU_VERSION: << parameters.cu_version >>
138

139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
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 >>

155
156
157
158
159
160
161
162
torchvision_android_params: &torchvision_android_params
  parameters:
    build_environment:
      type: string
      default: ""
  environment:
    BUILD_ENVIRONMENT: << parameters.build_environment >>

guyang3532's avatar
guyang3532 committed
163
164
165
166
167
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
    - image: torchvision/smoke_test:latest

168
jobs:
169
170
171
172
173
174
175
  circleci_consistency:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
176
            pip install --user --progress-bar off jinja2 pyyaml
177
178
179
            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)

180
181
182
183
184
185
186
  python_lint:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
187
188
189
            pip install --user --progress-bar off pre-commit
            pre-commit install-hooks
      - run: pre-commit run --all-files
190
191
192
      - run:
          name: Required lint modifications
          when: on_fail
193
          command: git --no-pager diff
194
195
196
197
198
199
200
201

  python_type_check:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
202
203
            sudo apt-get update -y
            sudo apt install -y libturbojpeg-dev
204
            pip install --user --progress-bar off mypy
205
            pip install --user --progress-bar off types-requests
206
            pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
207
            pip install --user --progress-bar off git+https://github.com/pytorch/data.git
208
            pip install --user --progress-bar off --no-build-isolation --editable .
209
            mypy --config-file mypy.ini
210

211
212
213
214
215
216
217
218
219
220
  docstring_parameters_sync:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
            pip install --user pydocstyle
            pydocstyle

221
222
223
224
225
226
227
  clang_format:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
228
229
            sudo apt-get update -y
            sudo apt install -y libtinfo5
230
231
232
            curl https://oss-clang-format.s3.us-east-2.amazonaws.com/linux64/clang-format-linux64 -o clang-format
            chmod +x clang-format
            sudo mv clang-format /opt/clang-format
233
            ./.circleci/unittest/linux/scripts/run-clang-format.py -r torchvision/csrc --clang-format-executable /opt/clang-format
234

Francisco Massa's avatar
Francisco Massa committed
235
236
237
238
239
240
241
242
243
  torchhub_test:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
            pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
            # need to install torchvision dependencies due to transitive imports
244
            pip install --user --progress-bar off --no-build-isolation .
Anirudh's avatar
Anirudh committed
245
            pip install pytest
Francisco Massa's avatar
Francisco Massa committed
246
247
            python test/test_hub.py

248
249
250
251
252
253
254
255
256
  torch_onnx_test:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
            pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
            # need to install torchvision dependencies due to transitive imports
257
            pip install --user --progress-bar off --no-build-isolation .
258
            pip install --user onnx
259
            pip install --user onnxruntime
260
            pip install --user pytest
261
262
            python test/test_onnx.py

263
264
265
  prototype_test:
    docker:
      - image: circleci/python:3.7
266
    resource_class: xlarge
267
268
269
    steps:
      - run:
          name: Install torch
270
271
          command: |
            pip install --user --progress-bar=off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
272
273
274
275
      - run:
          name: Install prototype dependencies
          command: pip install --user --progress-bar=off git+https://github.com/pytorch/data.git
      - checkout
276
277
278
279
280
      - run:
          name: Download model weights
          background: true
          command: |
            sudo apt update -qy && sudo apt install -qy parallel wget
281
            mkdir -p ~/.cache/torch/hub/checkpoints
282
            python scripts/collect_model_urls.py torchvision/prototype/models \
283
                | parallel -j0 'wget --no-verbose -O ~/.cache/torch/hub/checkpoints/`basename {}` {}\?source=ci'
284
285
286
287
      - run:
          name: Install torchvision
          command: pip install --user --progress-bar off --no-build-isolation .
      - run:
288
289
          name: Install test requirements
          command: pip install --user --progress-bar=off pytest pytest-mock scipy iopath
290
291
      - run:
          name: Run tests
292
293
          environment:
            PYTORCH_TEST_WITH_PROTOTYPE: 1
294
295
296
          command: pytest --junitxml=test-results/junit.xml -v --durations 20 test/test_prototype_*.py
      - store_test_results:
          path: test-results
297

298
299
300
  binary_linux_wheel:
    <<: *binary_common
    docker:
Edward Z. Yang's avatar
Edward Z. Yang committed
301
      - image: << parameters.wheel_docker_image >>
302
303
    resource_class: 2xlarge+
    steps:
304
      - checkout_merge
305
      - designate_upload_channel
306
307
308
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
309
310
311
312
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
313
314
315
316

  binary_linux_conda:
    <<: *binary_common
    docker:
317
      - image: "<< parameters.conda_docker_image >>"
318
    resource_class: 2xlarge+
319
    steps:
320
      - checkout_merge
321
      - designate_upload_channel
322
323
324
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
325
326
327
328
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"
329
330
      - store_test_results:
          path: build_results/
331

Francisco Massa's avatar
Francisco Massa committed
332
  binary_win_conda:
333
334
    <<: *binary_common
    executor: windows-cpu
335
336
    steps:
      - checkout_merge
337
      - designate_upload_channel
338
339
      - run:
          name: Build conda packages
340
          no_output_timeout: 20m
341
          command: |
342
343
344
345
346
347
            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"
348
349
350
351
            # cudatoolkit >= 11 isn't available for windows in the nvidia channel
            if [[ "${CU_VERSION}" =~ cu11.* ]]; then
              export CONDA_CHANNEL_FLAGS="-c conda-forge"
            fi
352
            packaging/build_conda.sh
353
            rm /C/tools/miniconda3/conda-bld/win-64/vs${VC_YEAR}*.tar.bz2
354
      - store_artifacts:
355
          path: C:/tools/miniconda3/conda-bld/win-64
356
      - persist_to_workspace:
357
          root: C:/tools/miniconda3/conda-bld/win-64
358
359
360
361
362
          paths:
            - "*"
      - store_test_results:
          path: build_results/

363
  binary_win_wheel:
364
365
    <<: *binary_common
    executor: windows-cpu
366
367
    steps:
      - checkout_merge
368
      - designate_upload_channel
369
370
371
      - run:
          name: Build wheel packages
          command: |
372
373
374
375
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_wheel.sh
376
      - store_artifacts:
377
          path: dist
378
      - persist_to_workspace:
379
          root: dist
380
381
382
383
384
          paths:
            - "*"
      - store_test_results:
          path: build_results/

385
386
387
  binary_macos_wheel:
    <<: *binary_common
    macos:
388
      xcode: "12.0"
389
    steps:
390
      - checkout_merge
391
      - designate_upload_channel
392
393
394
395
396
397
398
399
400
401
402
      - 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
403
404
405
406
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
407

408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
  binary_ios_build:
    <<: *torchvision_ios_params
    macos:
      xcode: "12.0"
    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:
      xcode: "12.0"
    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"

445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
  binary_android_build:
    <<: *torchvision_android_params
    docker:
      - image: circleci/android:api-29-ndk
    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:
      - image: circleci/android:api-29-ndk
    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"

481
482
483
  binary_macos_conda:
    <<: *binary_common
    macos:
484
      xcode: "12.0"
485
    steps:
486
      - checkout_merge
487
      - designate_upload_channel
488
489
490
491
492
493
494
495
496
      - 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
497
498
499
500
      - persist_to_workspace:
          root: /Users/distiller/miniconda3/conda-bld/osx-64
          paths:
            - "*"
501
502
      - store_test_results:
          path: build_results/
Edward Z. Yang's avatar
Edward Z. Yang committed
503
504
505
506
507
508
509
510

  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
511
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
512
513
514
515
516
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
517
            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
518
519
520

  # Requires org-member context
  binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
521
522
523
524
    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
525
526
527
528
529
    docker:
      - image: circleci/python:3.7
    steps:
      - attach_workspace:
          at: ~/workspace
530
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
531
532
533
534
535
536
537
538
539
540
      - checkout
      - run:
          command: |
            pip install --user awscli
            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
541
            for pkg in ~/workspace/*.whl; do
542
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
543
            done
544

guyang3532's avatar
guyang3532 committed
545
546
547
548
549
  smoke_test_linux_conda:
    <<: *smoke_test_common
    steps:
      - attach_workspace:
          at: ~/workspace
550
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
      - 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
569
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
      - run:
          name: install binaries
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
            pip install $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
      - 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:
      image: ubuntu-1604:201903-01
    resource_class: large
    environment:
      image_name: torchvision/smoke_test
    steps:
      - checkout
590
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
      - 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
610
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
611
612
613
614
615
616
      - 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
617
            conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
guyang3532's avatar
guyang3532 committed
618
            conda activate python${PYTHON_VERSION}
619
            conda install "Pillow>=5.3.0,!=8.3.*"
guyang3532's avatar
guyang3532 committed
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
            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
636
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
      - 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}
            pip install $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
      - run:
          name: smoke test
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate python${PYTHON_VERSION}
            python -c "import torchvision"

652
653
654
655
656
657
658
  unittest_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout
659
      - designate_upload_channel
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
      - 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:

          keys:
            - env-v2-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:

          key: env-v2-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

          paths:
            - conda
            - env
      - run:
          name: Install torchvision
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/linux/scripts/run_test.sh
      - run:
          name: Post process
          command: .circleci/unittest/linux/scripts/post_process.sh
      - store_test_results:
          path: test-results

  unittest_linux_gpu:
    <<: *binary_common
    machine:
694
      image: ubuntu-1604-cuda-10.2:202012-01
695
    resource_class: gpu.nvidia.medium
696
    environment:
697
      image_name: "pytorch/manylinux-cuda102"
698
      PYTHON_VERSION: << parameters.python_version >>
699
700
    steps:
      - checkout
701
      - designate_upload_channel
702
703
704
705
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
706
707
708
709
710
      - restore_cache:

          keys:
            - env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

711
712
      - run:
          name: Setup
713
          command: docker run -e PYTHON_VERSION -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/setup_env.sh
714
715
      - save_cache:

716
          key: env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
717
718
719
720

          paths:
            - conda
            - env
721
722
723
724
725
726
727
728
      - run:
          # Here we create an envlist file that contains some env variables that we want the docker container to be aware of.
          # Normally, the CIRCLECI variable is set and available on all CI workflows: https://circleci.com/docs/2.0/env-vars/#built-in-environment-variables.
          # They're avaiable in all the other workflows (OSX and Windows).
          # But here, we're running the unittest_linux_gpu workflows in a docker container, where those variables aren't accessible.
          # So instead we dump the variables we need in env.list and we pass that file when invoking "docker run".
          name: export CIRCLECI env var
          command: echo "CIRCLECI=true" >> ./env.list
729
730
      - run:
          name: Install torchvision
731
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e UPLOAD_CHANNEL -e CU_VERSION "${image_name}" .circleci/unittest/linux/scripts/install.sh
732
733
      - run:
          name: Run tests
734
          command: docker run --env-file ./env.list -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
735
736
737
738
739
740
741
742
743
744
745
746
      - run:
          name: Post Process
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/post_process.sh
      - store_test_results:
          path: test-results

  unittest_windows_cpu:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout
747
      - designate_upload_channel
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
      - 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:

          keys:
            - env-v2-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - save_cache:

          key: env-v2-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

          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:
784
      CUDA_VERSION: "11.1"
785
      PYTHON_VERSION: << parameters.python_version >>
786
787
    steps:
      - checkout
788
      - designate_upload_channel
789
790
791
792
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
793
794
795
796
      - restore_cache:

          keys:
            - env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
797

798
799
800
801
802
803
804
805
806
807
      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - save_cache:

          key: env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

          paths:
            - conda
            - env
808
809
810
811
812
813
      - run:
          name: Install CUDA
          command: packaging/windows/internal/cuda_install.bat
      - run:
          name: Update CUDA driver
          command: packaging/windows/internal/driver_update.bat
814
815
816
817
818
819
820
821
822
823
824
      - 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
825

Francisco Massa's avatar
Francisco Massa committed
826
827
828
  unittest_macos_cpu:
    <<: *binary_common
    macos:
829
      xcode: "12.0"
Francisco Massa's avatar
Francisco Massa committed
830
831
832
    resource_class: large
    steps:
      - checkout
833
      - designate_upload_channel
Francisco Massa's avatar
Francisco Massa committed
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
      - run:
          name: Install wget
          command: HOMEBREW_NO_AUTO_UPDATE=1 brew install wget
          # Disable brew auto update which is very slow
      - 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:

          keys:
            - env-v3-macos-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:

          key: env-v3-macos-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}

          paths:
            - conda
            - env
      - run:
          name: Install torchvision
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/linux/scripts/run_test.sh
      - run:
          name: Post process
          command: .circleci/unittest/linux/scripts/post_process.sh
      - store_test_results:
          path: test-results

869
870
871
872
873
874
875
  cmake_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout_merge
876
      - designate_upload_channel
877
878
879
880
881
882
883
884
      - run:
          name: Setup conda
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run: packaging/build_cmake.sh

  cmake_linux_gpu:
    <<: *binary_common
    machine:
885
      image: ubuntu-1604-cuda-10.2:202012-01
886
    resource_class: gpu.nvidia.small
887
888
889
890
891
892
893
    environment:
      PYTHON_VERSION: << parameters.python_version >>
      PYTORCH_VERSION: << parameters.pytorch_version >>
      UNICODE_ABI: << parameters.unicode_abi >>
      CU_VERSION: << parameters.cu_version >>
    steps:
      - checkout_merge
894
      - designate_upload_channel
895
896
897
898
899
      - 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
900
          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
901
902
903
904

  cmake_macos_cpu:
    <<: *binary_common
    macos:
905
      xcode: "12.0"
906
907
    steps:
      - checkout_merge
908
      - designate_upload_channel
909
910
911
912
913
914
915
916
917
918
919
920
921
922
      - 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 cmake
            packaging/build_cmake.sh

  cmake_windows_cpu:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout_merge
923
      - designate_upload_channel
924
925
926
927
928
929
930
931
932
933
934
935
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/build_cmake.sh

  cmake_windows_gpu:
    <<: *binary_common
    executor:
      name: windows-gpu
    steps:
      - checkout_merge
936
      - designate_upload_channel
937
938
939
940
941
942
943
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_cmake.sh

944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
  build_docs:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda100"
    resource_class: 2xlarge+
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - designate_upload_channel
      - run:
          name: Install torchvision
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Build docs
          command: |
            set -ex
964
965
            # turn v1.12.0rc3 into 1.12.0
            tag=$(echo $CIRCLE_TAG | sed -e 's/v*\([0-9.]*\).*/\1/')
966
            VERSION=${tag:-main}
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
            eval "$(./conda/bin/conda shell.bash hook)"
            conda activate ./env
            pushd docs
            pip install -r requirements.txt
            make html
            popd
      - persist_to_workspace:
          root: ./
          paths:
            - "*"
      - store_artifacts:
          path: ./docs/build/html
          destination: docs

  upload_docs:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda100"
    resource_class: 2xlarge+
    steps:
      - attach_workspace:
          at: ~/workspace
      - run:
          name: Generate netrc
          command: |
            # set credentials for https pushing
            # requires the org-member context
            cat > ~/.netrc \<<DONE
              machine github.com
              login pytorchbot
              password ${GITHUB_PYTORCHBOT_TOKEN}
            DONE
      - run:
          name: Upload docs
          command: |
            # Don't use "checkout" step since it uses ssh, which cannot git push
            # https://circleci.com/docs/2.0/configuration-reference/#checkout
            set -ex
            tag=${CIRCLE_TAG:1:5}
1006
            target=${tag:-main}
1007
1008
1009
            ~/workspace/.circleci/build_docs/commit_docs.sh ~/workspace $target


1010
1011
1012
1013
workflows:
  build:
    jobs:
      - circleci_consistency
Edward Z. Yang's avatar
Edward Z. Yang committed
1014
      - binary_linux_wheel:
1015
          conda_docker_image: pytorch/conda-builder:cpu
1016
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1017
          name: binary_linux_wheel_py3.6_cpu
1018
          python_version: '3.6'
1019
1020
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1021
          conda_docker_image: pytorch/conda-builder:cuda102
1022
1023
1024
1025
          cu_version: cu102
          name: binary_linux_wheel_py3.6_cu102
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1026
      - binary_linux_wheel:
1027
1028
1029
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1030
          python_version: '3.6'
1031
          wheel_docker_image: pytorch/manylinux-cuda111
1032
1033
1034
1035
1036
1037
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_wheel_py3.6_cu113
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda113
1038
1039
1040
1041
1042
      - binary_linux_wheel:
          cu_version: rocm4.2
          name: binary_linux_wheel_py3.6_rocm4.2
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1043
1044
1045
1046
1047
      - binary_linux_wheel:
          cu_version: rocm4.3.1
          name: binary_linux_wheel_py3.6_rocm4.3.1
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
Edward Z. Yang's avatar
Edward Z. Yang committed
1048
      - binary_linux_wheel:
1049
          conda_docker_image: pytorch/conda-builder:cpu
1050
          cu_version: cpu
1051
1052
          filters:
            branches:
1053
              only: /.*/
1054
1055
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
1056
          name: binary_linux_wheel_py3.7_cpu
1057
          python_version: '3.7'
1058
1059
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1060
          conda_docker_image: pytorch/conda-builder:cuda102
1061
1062
1063
1064
          cu_version: cu102
          name: binary_linux_wheel_py3.7_cu102
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1065
      - binary_linux_wheel:
1066
1067
1068
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1069
          python_version: '3.7'
1070
          wheel_docker_image: pytorch/manylinux-cuda111
1071
1072
1073
1074
1075
1076
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_wheel_py3.7_cu113
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda113
1077
1078
1079
1080
1081
      - binary_linux_wheel:
          cu_version: rocm4.2
          name: binary_linux_wheel_py3.7_rocm4.2
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1082
1083
1084
1085
1086
      - binary_linux_wheel:
          cu_version: rocm4.3.1
          name: binary_linux_wheel_py3.7_rocm4.3.1
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
1087
      - binary_linux_wheel:
1088
          conda_docker_image: pytorch/conda-builder:cpu
1089
1090
1091
          cu_version: cpu
          name: binary_linux_wheel_py3.8_cpu
          python_version: '3.8'
1092
1093
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1094
          conda_docker_image: pytorch/conda-builder:cuda102
1095
1096
1097
1098
          cu_version: cu102
          name: binary_linux_wheel_py3.8_cu102
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1099
      - binary_linux_wheel:
1100
1101
1102
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1103
          python_version: '3.8'
1104
          wheel_docker_image: pytorch/manylinux-cuda111
1105
1106
1107
1108
1109
1110
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_wheel_py3.8_cu113
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda113
1111
1112
1113
1114
1115
      - binary_linux_wheel:
          cu_version: rocm4.2
          name: binary_linux_wheel_py3.8_rocm4.2
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1116
1117
1118
1119
1120
      - binary_linux_wheel:
          cu_version: rocm4.3.1
          name: binary_linux_wheel_py3.8_rocm4.3.1
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          name: binary_linux_wheel_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
          name: binary_linux_wheel_py3.9_cu102
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1134
1135
1136
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.9_cu111
1137
          python_version: '3.9'
1138
          wheel_docker_image: pytorch/manylinux-cuda111
1139
1140
1141
1142
1143
1144
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_wheel_py3.9_cu113
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda113
1145
1146
1147
1148
1149
      - binary_linux_wheel:
          cu_version: rocm4.2
          name: binary_linux_wheel_py3.9_rocm4.2
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1150
1151
1152
1153
1154
      - binary_linux_wheel:
          cu_version: rocm4.3.1
          name: binary_linux_wheel_py3.9_rocm4.3.1
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
Edward Z. Yang's avatar
Edward Z. Yang committed
1155
      - binary_macos_wheel:
1156
          conda_docker_image: pytorch/conda-builder:cpu
1157
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1158
          name: binary_macos_wheel_py3.6_cpu
1159
          python_version: '3.6'
1160
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1161
      - binary_macos_wheel:
1162
          conda_docker_image: pytorch/conda-builder:cpu
1163
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1164
          name: binary_macos_wheel_py3.7_cpu
1165
          python_version: '3.7'
1166
          wheel_docker_image: pytorch/manylinux-cuda102
1167
      - binary_macos_wheel:
1168
          conda_docker_image: pytorch/conda-builder:cpu
1169
1170
1171
          cu_version: cpu
          name: binary_macos_wheel_py3.8_cpu
          python_version: '3.8'
1172
          wheel_docker_image: pytorch/manylinux-cuda102
1173
1174
1175
1176
1177
1178
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          name: binary_macos_wheel_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
1179
      - binary_win_wheel:
1180
1181
1182
          cu_version: cpu
          filters:
            branches:
1183
              only: main
1184
1185
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1186
1187
          name: binary_win_wheel_py3.6_cpu
          python_version: '3.6'
peterjc123's avatar
peterjc123 committed
1188
      - binary_win_wheel:
1189
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1190
1191
          filters:
            branches:
1192
              only: main
peterjc123's avatar
peterjc123 committed
1193
1194
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1195
          name: binary_win_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1196
          python_version: '3.6'
1197
1198
1199
1200
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
1201
              only: main
1202
1203
1204
1205
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_wheel_py3.6_cu113
          python_version: '3.6'
1206
      - binary_win_wheel:
1207
1208
1209
          cu_version: cpu
          filters:
            branches:
1210
              only: main
1211
1212
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1213
1214
          name: binary_win_wheel_py3.7_cpu
          python_version: '3.7'
peterjc123's avatar
peterjc123 committed
1215
      - binary_win_wheel:
1216
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1217
1218
          filters:
            branches:
1219
              only: main
peterjc123's avatar
peterjc123 committed
1220
1221
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1222
          name: binary_win_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1223
          python_version: '3.7'
1224
1225
1226
1227
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
1228
              only: main
1229
1230
1231
1232
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_wheel_py3.7_cu113
          python_version: '3.7'
1233
      - binary_win_wheel:
1234
          cu_version: cpu
1235
1236
          filters:
            branches:
1237
              only: main
1238
1239
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1240
1241
          name: binary_win_wheel_py3.8_cpu
          python_version: '3.8'
peterjc123's avatar
peterjc123 committed
1242
      - binary_win_wheel:
1243
          cu_version: cu111
1244
1245
          filters:
            branches:
1246
              only: main
1247
1248
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1249
          name: binary_win_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1250
          python_version: '3.8'
1251
1252
1253
1254
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
1255
              only: main
1256
1257
1258
1259
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_wheel_py3.8_cu113
          python_version: '3.8'
1260
1261
1262
1263
1264
      - binary_win_wheel:
          cu_version: cpu
          name: binary_win_wheel_py3.9_cpu
          python_version: '3.9'
      - binary_win_wheel:
1265
          cu_version: cu111
1266
1267
          filters:
            branches:
1268
              only: main
1269
1270
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1271
          name: binary_win_wheel_py3.9_cu111
1272
          python_version: '3.9'
1273
1274
1275
1276
      - binary_win_wheel:
          cu_version: cu113
          name: binary_win_wheel_py3.9_cu113
          python_version: '3.9'
Edward Z. Yang's avatar
Edward Z. Yang committed
1277
      - binary_linux_conda:
1278
          conda_docker_image: pytorch/conda-builder:cpu
1279
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1280
          name: binary_linux_conda_py3.6_cpu
1281
          python_version: '3.6'
1282
1283
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1284
          conda_docker_image: pytorch/conda-builder:cuda102
1285
1286
1287
1288
          cu_version: cu102
          name: binary_linux_conda_py3.6_cu102
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1289
      - binary_linux_conda:
1290
1291
1292
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1293
          python_version: '3.6'
1294
          wheel_docker_image: pytorch/manylinux-cuda111
1295
1296
1297
1298
1299
1300
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_conda_py3.6_cu113
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda113
Edward Z. Yang's avatar
Edward Z. Yang committed
1301
      - binary_linux_conda:
1302
          conda_docker_image: pytorch/conda-builder:cpu
1303
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1304
          name: binary_linux_conda_py3.7_cpu
1305
          python_version: '3.7'
1306
1307
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1308
          conda_docker_image: pytorch/conda-builder:cuda102
1309
1310
1311
1312
          cu_version: cu102
          name: binary_linux_conda_py3.7_cu102
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1313
      - binary_linux_conda:
1314
1315
1316
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1317
          python_version: '3.7'
1318
          wheel_docker_image: pytorch/manylinux-cuda111
1319
1320
1321
1322
1323
1324
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_conda_py3.7_cu113
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda113
1325
      - binary_linux_conda:
1326
          conda_docker_image: pytorch/conda-builder:cpu
1327
1328
1329
          cu_version: cpu
          name: binary_linux_conda_py3.8_cpu
          python_version: '3.8'
1330
1331
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1332
          conda_docker_image: pytorch/conda-builder:cuda102
1333
1334
1335
1336
          cu_version: cu102
          name: binary_linux_conda_py3.8_cu102
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
1337
      - binary_linux_conda:
1338
1339
1340
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1341
          python_version: '3.8'
1342
          wheel_docker_image: pytorch/manylinux-cuda111
1343
1344
1345
1346
1347
1348
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_conda_py3.8_cu113
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda113
1349
1350
1351
1352
1353
1354
1355
1356
1357
1358
1359
1360
1361
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          name: binary_linux_conda_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
          name: binary_linux_conda_py3.9_cu102
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1362
1363
1364
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.9_cu111
1365
          python_version: '3.9'
1366
          wheel_docker_image: pytorch/manylinux-cuda111
1367
1368
1369
1370
1371
1372
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          name: binary_linux_conda_py3.9_cu113
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda113
Edward Z. Yang's avatar
Edward Z. Yang committed
1373
      - binary_macos_conda:
1374
          conda_docker_image: pytorch/conda-builder:cpu
1375
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1376
          name: binary_macos_conda_py3.6_cpu
1377
          python_version: '3.6'
1378
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1379
      - binary_macos_conda:
1380
          conda_docker_image: pytorch/conda-builder:cpu
1381
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1382
          name: binary_macos_conda_py3.7_cpu
1383
          python_version: '3.7'
1384
          wheel_docker_image: pytorch/manylinux-cuda102
1385
      - binary_macos_conda:
1386
          conda_docker_image: pytorch/conda-builder:cpu
1387
1388
1389
          cu_version: cpu
          name: binary_macos_conda_py3.8_cpu
          python_version: '3.8'
1390
          wheel_docker_image: pytorch/manylinux-cuda102
1391
1392
1393
1394
1395
1396
      - binary_macos_conda:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          name: binary_macos_conda_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
1397
      - binary_win_conda:
1398
1399
1400
          cu_version: cpu
          filters:
            branches:
1401
              only: main
1402
1403
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1404
1405
          name: binary_win_conda_py3.6_cpu
          python_version: '3.6'
peterjc123's avatar
peterjc123 committed
1406
      - binary_win_conda:
1407
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1408
1409
          filters:
            branches:
1410
              only: main
peterjc123's avatar
peterjc123 committed
1411
1412
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1413
          name: binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1414
          python_version: '3.6'
1415
1416
1417
1418
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
1419
              only: main
1420
1421
1422
1423
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_conda_py3.6_cu113
          python_version: '3.6'
1424
      - binary_win_conda:
1425
1426
1427
          cu_version: cpu
          filters:
            branches:
1428
              only: main
1429
1430
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1431
1432
          name: binary_win_conda_py3.7_cpu
          python_version: '3.7'
peterjc123's avatar
peterjc123 committed
1433
      - binary_win_conda:
1434
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1435
1436
          filters:
            branches:
1437
              only: main
peterjc123's avatar
peterjc123 committed
1438
1439
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1440
          name: binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1441
          python_version: '3.7'
1442
1443
1444
1445
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
1446
              only: main
1447
1448
1449
1450
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_conda_py3.7_cu113
          python_version: '3.7'
1451
      - binary_win_conda:
1452
          cu_version: cpu
1453
1454
          filters:
            branches:
1455
              only: main
1456
1457
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1458
1459
          name: binary_win_conda_py3.8_cpu
          python_version: '3.8'
peterjc123's avatar
peterjc123 committed
1460
      - binary_win_conda:
1461
          cu_version: cu111
1462
1463
          filters:
            branches:
1464
              only: main
1465
1466
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1467
          name: binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1468
          python_version: '3.8'
1469
1470
1471
1472
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
1473
              only: main
1474
1475
1476
1477
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_conda_py3.8_cu113
          python_version: '3.8'
1478
1479
1480
1481
1482
      - binary_win_conda:
          cu_version: cpu
          name: binary_win_conda_py3.9_cpu
          python_version: '3.9'
      - binary_win_conda:
1483
          cu_version: cu111
1484
1485
          filters:
            branches:
1486
              only: main
1487
1488
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1489
          name: binary_win_conda_py3.9_cu111
1490
          python_version: '3.9'
1491
1492
1493
1494
      - binary_win_conda:
          cu_version: cu113
          name: binary_win_conda_py3.9_cu113
          python_version: '3.9'
1495
      - build_docs:
1496
1497
1498
          filters:
            branches:
              only:
1499
              - /.*/
1500
1501
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
          name: build_docs
          python_version: '3.7'
          requires:
          - binary_linux_wheel_py3.7_cpu
      - upload_docs:
          context: org-member
          filters:
            branches:
              only:
              - nightly
1512
1513
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1514
1515
1516
1517
          name: upload_docs
          python_version: '3.7'
          requires:
          - build_docs
1518
      - python_lint
1519
      - python_type_check
1520
      - docstring_parameters_sync
1521
      - clang_format
Francisco Massa's avatar
Francisco Massa committed
1522
      - torchhub_test
1523
      - torch_onnx_test
1524
      - prototype_test
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
      - binary_ios_build:
          build_environment: binary-libtorchvision_ops-ios-12.0.0-x86_64
          ios_arch: x86_64
          ios_platform: SIMULATOR
          name: binary_libtorchvision_ops_ios_12.0.0_x86_64
      - binary_ios_build:
          build_environment: binary-libtorchvision_ops-ios-12.0.0-arm64
          ios_arch: arm64
          ios_platform: OS
          name: binary_libtorchvision_ops_ios_12.0.0_arm64
1535
1536
1537
      - binary_android_build:
          build_environment: binary-libtorchvision_ops-android
          name: binary_libtorchvision_ops_android
Edward Z. Yang's avatar
Edward Z. Yang committed
1538

1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
1550
1551
1552
  unittest:
    jobs:
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.6
          python_version: '3.6'
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.7
          python_version: '3.7'
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.8
          python_version: '3.8'
1553
1554
1555
1556
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.9
          python_version: '3.9'
1557
      - unittest_linux_gpu:
1558
          cu_version: cu102
1559
1560
1561
          filters:
            branches:
              only:
1562
              - main
1563
1564
1565
1566
              - nightly
          name: unittest_linux_gpu_py3.6
          python_version: '3.6'
      - unittest_linux_gpu:
1567
          cu_version: cu102
1568
1569
1570
          filters:
            branches:
              only:
1571
              - main
1572
1573
1574
1575
              - nightly
          name: unittest_linux_gpu_py3.7
          python_version: '3.7'
      - unittest_linux_gpu:
1576
          cu_version: cu102
1577
1578
          name: unittest_linux_gpu_py3.8
          python_version: '3.8'
1579
      - unittest_linux_gpu:
1580
          cu_version: cu102
1581
1582
1583
          filters:
            branches:
              only:
1584
              - main
1585
1586
1587
              - nightly
          name: unittest_linux_gpu_py3.9
          python_version: '3.9'
1588
1589
1590
1591
1592
1593
1594
1595
1596
1597
1598
1599
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.6
          python_version: '3.6'
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.7
          python_version: '3.7'
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.8
          python_version: '3.8'
1600
1601
1602
1603
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.9
          python_version: '3.9'
1604
      - unittest_windows_gpu:
1605
          cu_version: cu102
1606
1607
1608
          filters:
            branches:
              only:
1609
              - main
1610
1611
1612
1613
              - nightly
          name: unittest_windows_gpu_py3.6
          python_version: '3.6'
      - unittest_windows_gpu:
1614
          cu_version: cu102
1615
1616
1617
          filters:
            branches:
              only:
1618
              - main
1619
1620
1621
1622
              - nightly
          name: unittest_windows_gpu_py3.7
          python_version: '3.7'
      - unittest_windows_gpu:
1623
          cu_version: cu102
1624
1625
          name: unittest_windows_gpu_py3.8
          python_version: '3.8'
1626
      - unittest_windows_gpu:
1627
          cu_version: cu102
1628
1629
1630
          filters:
            branches:
              only:
1631
              - main
1632
1633
1634
              - nightly
          name: unittest_windows_gpu_py3.9
          python_version: '3.9'
Francisco Massa's avatar
Francisco Massa committed
1635
1636
1637
1638
1639
1640
1641
1642
1643
1644
1645
1646
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.6
          python_version: '3.6'
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.7
          python_version: '3.7'
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.8
          python_version: '3.8'
1647
1648
1649
1650
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.9
          python_version: '3.9'
1651
1652
1653
1654
1655
1656
1657
1658

  cmake:
    jobs:
      - cmake_linux_cpu:
          cu_version: cpu
          name: cmake_linux_cpu
          python_version: '3.8'
      - cmake_linux_gpu:
1659
          cu_version: cu113
1660
1661
          name: cmake_linux_gpu
          python_version: '3.8'
1662
          wheel_docker_image: pytorch/manylinux-cuda113
1663
1664
1665
1666
      - cmake_windows_cpu:
          cu_version: cpu
          name: cmake_windows_cpu
          python_version: '3.8'
1667
      - cmake_windows_gpu:
1668
          cu_version: cu113
1669
1670
          name: cmake_windows_gpu
          python_version: '3.8'
1671
1672
1673
1674
1675
      - cmake_macos_cpu:
          cu_version: cpu
          name: cmake_macos_cpu
          python_version: '3.8'

Edward Z. Yang's avatar
Edward Z. Yang committed
1676
1677
1678
  nightly:
    jobs:
      - circleci_consistency
1679
      - python_lint
1680
      - python_type_check
1681
      - docstring_parameters_sync
1682
      - clang_format
Francisco Massa's avatar
Francisco Massa committed
1683
      - torchhub_test
1684
      - torch_onnx_test
1685
      - prototype_test
1686
1687
1688
1689
1690
1691
1692
1693
1694
1695
1696
1697
1698
1699
1700
1701
1702
1703
1704
1705
1706
1707
1708
1709
1710
1711
1712
1713
      - binary_ios_build:
          build_environment: nightly-binary-libtorchvision_ops-ios-12.0.0-x86_64
          filters:
            branches:
              only:
              - nightly
          ios_arch: x86_64
          ios_platform: SIMULATOR
          name: nightly_binary_libtorchvision_ops_ios_12.0.0_x86_64
      - binary_ios_build:
          build_environment: nightly-binary-libtorchvision_ops-ios-12.0.0-arm64
          filters:
            branches:
              only:
              - nightly
          ios_arch: arm64
          ios_platform: OS
          name: nightly_binary_libtorchvision_ops_ios_12.0.0_arm64
      - binary_ios_upload:
          build_environment: nightly-binary-libtorchvision_ops-ios-12.0.0-upload
          context: org-member
          filters:
            branches:
              only:
              - nightly
          requires:
          - nightly_binary_libtorchvision_ops_ios_12.0.0_x86_64
          - nightly_binary_libtorchvision_ops_ios_12.0.0_arm64
1714
1715
1716
1717
1718
1719
1720
1721
      - binary_android_upload:
          build_environment: nightly-binary-libtorchvision_ops-android-upload
          context: org-member
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_libtorchvision_ops_android_upload
1722
      - binary_linux_wheel:
1723
          conda_docker_image: pytorch/conda-builder:cpu
1724
          cu_version: cpu
1725
1726
1727
          filters:
            branches:
              only: nightly
1728
1729
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
1730
          name: nightly_binary_linux_wheel_py3.6_cpu
1731
          python_version: '3.6'
1732
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1733
1734
      - binary_wheel_upload:
          context: org-member
1735
1736
1737
          filters:
            branches:
              only: nightly
1738
1739
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1740
          name: nightly_binary_linux_wheel_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
1741
          requires:
1742
1743
          - nightly_binary_linux_wheel_py3.6_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
1744
1745
1746
1747
1748
1749
1750
1751
1752
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.6_cpu_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_linux_wheel_py3.6_cpu_upload
Francisco Massa's avatar
Francisco Massa committed
1753
      - binary_linux_wheel:
1754
          conda_docker_image: pytorch/conda-builder:cuda102
1755
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
1756
1757
1758
          filters:
            branches:
              only: nightly
1759
1760
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1761
          name: nightly_binary_linux_wheel_py3.6_cu102
Francisco Massa's avatar
Francisco Massa committed
1762
          python_version: '3.6'
1763
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
1764
1765
1766
1767
1768
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1769
1770
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1771
          name: nightly_binary_linux_wheel_py3.6_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
1772
          requires:
1773
1774
          - nightly_binary_linux_wheel_py3.6_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
1775
1776
1777
1778
1779
1780
1781
1782
1783
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.6_cu102_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_linux_wheel_py3.6_cu102_upload
peterjc123's avatar
peterjc123 committed
1784
      - binary_linux_wheel:
1785
1786
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1787
1788
1789
1790
1791
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1792
          name: nightly_binary_linux_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1793
          python_version: '3.6'
1794
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
1795
1796
1797
1798
1799
1800
1801
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1802
          name: nightly_binary_linux_wheel_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
1803
          requires:
1804
1805
          - nightly_binary_linux_wheel_py3.6_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
1806
1807
1808
1809
1810
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1811
          name: nightly_binary_linux_wheel_py3.6_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
1812
1813
          python_version: '3.6'
          requires:
1814
          - nightly_binary_linux_wheel_py3.6_cu111_upload
1815
1816
1817
1818
1819
1820
1821
1822
1823
1824
1825
1826
1827
1828
1829
1830
1831
1832
1833
1834
1835
1836
1837
1838
1839
1840
1841
1842
1843
1844
1845
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.6_cu113
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.6_cu113_upload
          requires:
          - nightly_binary_linux_wheel_py3.6_cu113
          subfolder: cu113/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.6_cu113_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_linux_wheel_py3.6_cu113_upload
Jeff Daily's avatar
Jeff Daily committed
1846
      - binary_linux_wheel:
1847
          cu_version: rocm4.2
Jeff Daily's avatar
Jeff Daily committed
1848
1849
1850
1851
1852
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1853
          name: nightly_binary_linux_wheel_py3.6_rocm4.2
Jeff Daily's avatar
Jeff Daily committed
1854
          python_version: '3.6'
1855
          wheel_docker_image: pytorch/manylinux-rocm:4.2
Jeff Daily's avatar
Jeff Daily committed
1856
1857
1858
1859
1860
1861
1862
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1863
          name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
Jeff Daily's avatar
Jeff Daily committed
1864
          requires:
1865
1866
          - nightly_binary_linux_wheel_py3.6_rocm4.2
          subfolder: rocm4.2/
Jeff Daily's avatar
Jeff Daily committed
1867
1868
1869
1870
1871
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1872
          name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
1873
1874
          python_version: '3.6'
          requires:
1875
          - nightly_binary_linux_wheel_py3.6_rocm4.2_upload
1876
      - binary_linux_wheel:
1877
          cu_version: rocm4.3.1
1878
1879
1880
1881
1882
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1883
          name: nightly_binary_linux_wheel_py3.6_rocm4.3.1
1884
          python_version: '3.6'
1885
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
1886
1887
1888
1889
1890
1891
1892
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1893
          name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
1894
          requires:
1895
1896
          - nightly_binary_linux_wheel_py3.6_rocm4.3.1
          subfolder: rocm4.3.1/
1897
1898
1899
1900
1901
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1902
          name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_smoke_test_pip
1903
1904
          python_version: '3.6'
          requires:
1905
          - nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
1906
      - binary_linux_wheel:
1907
          conda_docker_image: pytorch/conda-builder:cpu
1908
          cu_version: cpu
1909
1910
1911
          filters:
            branches:
              only: nightly
1912
1913
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
1914
          name: nightly_binary_linux_wheel_py3.7_cpu
1915
          python_version: '3.7'
1916
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1917
1918
      - binary_wheel_upload:
          context: org-member
1919
1920
1921
          filters:
            branches:
              only: nightly
1922
1923
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1924
          name: nightly_binary_linux_wheel_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
1925
          requires:
1926
1927
          - nightly_binary_linux_wheel_py3.7_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
1928
1929
1930
1931
1932
1933
1934
1935
1936
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.7_cpu_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_linux_wheel_py3.7_cpu_upload
Francisco Massa's avatar
Francisco Massa committed
1937
      - binary_linux_wheel:
1938
          conda_docker_image: pytorch/conda-builder:cuda102
1939
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
1940
1941
1942
          filters:
            branches:
              only: nightly
1943
1944
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1945
          name: nightly_binary_linux_wheel_py3.7_cu102
Francisco Massa's avatar
Francisco Massa committed
1946
          python_version: '3.7'
1947
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
1948
1949
1950
1951
1952
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1953
1954
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1955
          name: nightly_binary_linux_wheel_py3.7_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
1956
          requires:
1957
1958
          - nightly_binary_linux_wheel_py3.7_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
1959
1960
1961
1962
1963
1964
1965
1966
1967
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.7_cu102_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_linux_wheel_py3.7_cu102_upload
peterjc123's avatar
peterjc123 committed
1968
      - binary_linux_wheel:
1969
1970
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1971
1972
1973
1974
1975
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1976
          name: nightly_binary_linux_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1977
          python_version: '3.7'
1978
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
1979
1980
1981
1982
1983
1984
1985
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1986
          name: nightly_binary_linux_wheel_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
1987
          requires:
1988
1989
          - nightly_binary_linux_wheel_py3.7_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
1990
1991
1992
1993
1994
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1995
          name: nightly_binary_linux_wheel_py3.7_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
1996
1997
          python_version: '3.7'
          requires:
1998
          - nightly_binary_linux_wheel_py3.7_cu111_upload
1999
2000
2001
2002
2003
2004
2005
2006
2007
2008
2009
2010
2011
2012
2013
2014
2015
2016
2017
2018
2019
2020
2021
2022
2023
2024
2025
2026
2027
2028
2029
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.7_cu113
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.7_cu113_upload
          requires:
          - nightly_binary_linux_wheel_py3.7_cu113
          subfolder: cu113/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.7_cu113_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_linux_wheel_py3.7_cu113_upload
Jeff Daily's avatar
Jeff Daily committed
2030
      - binary_linux_wheel:
2031
          cu_version: rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2032
2033
2034
2035
2036
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2037
          name: nightly_binary_linux_wheel_py3.7_rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2038
          python_version: '3.7'
2039
          wheel_docker_image: pytorch/manylinux-rocm:4.2
Jeff Daily's avatar
Jeff Daily committed
2040
2041
2042
2043
2044
2045
2046
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2047
          name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
Jeff Daily's avatar
Jeff Daily committed
2048
          requires:
2049
2050
          - nightly_binary_linux_wheel_py3.7_rocm4.2
          subfolder: rocm4.2/
Jeff Daily's avatar
Jeff Daily committed
2051
2052
2053
2054
2055
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2056
          name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
2057
2058
          python_version: '3.7'
          requires:
2059
          - nightly_binary_linux_wheel_py3.7_rocm4.2_upload
2060
      - binary_linux_wheel:
2061
          cu_version: rocm4.3.1
2062
2063
2064
2065
2066
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2067
          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
2068
          python_version: '3.7'
2069
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
2070
2071
2072
2073
2074
2075
2076
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2077
          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
2078
          requires:
2079
2080
          - nightly_binary_linux_wheel_py3.7_rocm4.3.1
          subfolder: rocm4.3.1/
2081
2082
2083
2084
2085
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2086
          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
2087
2088
          python_version: '3.7'
          requires:
2089
          - nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
2090
      - binary_linux_wheel:
2091
          conda_docker_image: pytorch/conda-builder:cpu
2092
2093
2094
2095
          cu_version: cpu
          filters:
            branches:
              only: nightly
2096
2097
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2098
2099
          name: nightly_binary_linux_wheel_py3.8_cpu
          python_version: '3.8'
2100
          wheel_docker_image: pytorch/manylinux-cuda102
2101
2102
2103
2104
2105
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2106
2107
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2108
2109
2110
2111
          name: nightly_binary_linux_wheel_py3.8_cpu_upload
          requires:
          - nightly_binary_linux_wheel_py3.8_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
2112
2113
2114
2115
2116
2117
2118
2119
2120
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_linux_wheel_py3.8_cpu_upload
2121
      - binary_linux_wheel:
2122
          conda_docker_image: pytorch/conda-builder:cuda102
2123
          cu_version: cu102
2124
2125
2126
          filters:
            branches:
              only: nightly
2127
2128
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2129
          name: nightly_binary_linux_wheel_py3.8_cu102
2130
          python_version: '3.8'
2131
          wheel_docker_image: pytorch/manylinux-cuda102
2132
2133
2134
2135
2136
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2137
2138
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2139
          name: nightly_binary_linux_wheel_py3.8_cu102_upload
2140
          requires:
2141
2142
          - nightly_binary_linux_wheel_py3.8_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
2143
2144
2145
2146
2147
2148
2149
2150
2151
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.8_cu102_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_linux_wheel_py3.8_cu102_upload
peterjc123's avatar
peterjc123 committed
2152
      - binary_linux_wheel:
2153
2154
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2155
2156
2157
2158
2159
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2160
          name: nightly_binary_linux_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
2161
          python_version: '3.8'
2162
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
2163
2164
2165
2166
2167
2168
2169
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2170
          name: nightly_binary_linux_wheel_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
2171
          requires:
2172
2173
          - nightly_binary_linux_wheel_py3.8_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2174
2175
2176
2177
2178
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2179
          name: nightly_binary_linux_wheel_py3.8_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2180
2181
          python_version: '3.8'
          requires:
2182
          - nightly_binary_linux_wheel_py3.8_cu111_upload
2183
2184
2185
2186
2187
2188
2189
2190
2191
2192
2193
2194
2195
2196
2197
2198
2199
2200
2201
2202
2203
2204
2205
2206
2207
2208
2209
2210
2211
2212
2213
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.8_cu113
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.8_cu113_upload
          requires:
          - nightly_binary_linux_wheel_py3.8_cu113
          subfolder: cu113/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.8_cu113_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_linux_wheel_py3.8_cu113_upload
Jeff Daily's avatar
Jeff Daily committed
2214
      - binary_linux_wheel:
2215
          cu_version: rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2216
2217
2218
2219
2220
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2221
          name: nightly_binary_linux_wheel_py3.8_rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2222
          python_version: '3.8'
2223
          wheel_docker_image: pytorch/manylinux-rocm:4.2
Jeff Daily's avatar
Jeff Daily committed
2224
2225
2226
2227
2228
2229
2230
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2231
          name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
Jeff Daily's avatar
Jeff Daily committed
2232
          requires:
2233
2234
          - nightly_binary_linux_wheel_py3.8_rocm4.2
          subfolder: rocm4.2/
Jeff Daily's avatar
Jeff Daily committed
2235
2236
2237
2238
2239
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2240
          name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
2241
2242
          python_version: '3.8'
          requires:
2243
          - nightly_binary_linux_wheel_py3.8_rocm4.2_upload
2244
      - binary_linux_wheel:
2245
          cu_version: rocm4.3.1
2246
2247
2248
2249
2250
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2251
          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
2252
          python_version: '3.8'
2253
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
2254
2255
2256
2257
2258
2259
2260
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2261
          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
2262
          requires:
2263
2264
          - nightly_binary_linux_wheel_py3.8_rocm4.3.1
          subfolder: rocm4.3.1/
2265
2266
2267
2268
2269
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2270
          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
2271
2272
          python_version: '3.8'
          requires:
2273
          - nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
2274
2275
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
2276
          cu_version: cpu
2277
2278
2279
          filters:
            branches:
              only: nightly
2280
2281
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2282
2283
          name: nightly_binary_linux_wheel_py3.9_cpu
          python_version: '3.9'
2284
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
2285
2286
      - binary_wheel_upload:
          context: org-member
2287
2288
2289
          filters:
            branches:
              only: nightly
2290
2291
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2292
          name: nightly_binary_linux_wheel_py3.9_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2293
          requires:
2294
2295
2296
2297
2298
2299
2300
2301
2302
2303
2304
2305
2306
2307
          - nightly_binary_linux_wheel_py3.9_cpu
          subfolder: cpu/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.9_cpu_smoke_test_pip
          python_version: '3.9'
          requires:
          - nightly_binary_linux_wheel_py3.9_cpu_upload
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
2308
2309
2310
          filters:
            branches:
              only: nightly
2311
2312
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2313
2314
          name: nightly_binary_linux_wheel_py3.9_cu102
          python_version: '3.9'
2315
          wheel_docker_image: pytorch/manylinux-cuda102
2316
2317
2318
2319
2320
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2321
2322
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2323
          name: nightly_binary_linux_wheel_py3.9_cu102_upload
2324
          requires:
2325
2326
2327
2328
2329
2330
2331
2332
2333
2334
2335
2336
          - nightly_binary_linux_wheel_py3.9_cu102
          subfolder: cu102/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.9_cu102_smoke_test_pip
          python_version: '3.9'
          requires:
          - nightly_binary_linux_wheel_py3.9_cu102_upload
      - binary_linux_wheel:
2337
2338
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
2339
2340
2341
          filters:
            branches:
              only: nightly
2342
2343
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2344
          name: nightly_binary_linux_wheel_py3.9_cu111
2345
          python_version: '3.9'
2346
          wheel_docker_image: pytorch/manylinux-cuda111
2347
2348
2349
2350
2351
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2352
2353
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2354
          name: nightly_binary_linux_wheel_py3.9_cu111_upload
2355
          requires:
2356
2357
          - nightly_binary_linux_wheel_py3.9_cu111
          subfolder: cu111/
2358
      - smoke_test_linux_pip:
guyang3532's avatar
guyang3532 committed
2359
2360
2361
2362
          filters:
            branches:
              only:
              - nightly
2363
          name: nightly_binary_linux_wheel_py3.9_cu111_smoke_test_pip
2364
          python_version: '3.9'
guyang3532's avatar
guyang3532 committed
2365
          requires:
2366
          - nightly_binary_linux_wheel_py3.9_cu111_upload
2367
2368
2369
2370
2371
2372
2373
2374
2375
2376
2377
2378
2379
2380
2381
2382
2383
2384
2385
2386
2387
2388
2389
2390
2391
2392
2393
2394
2395
2396
2397
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.9_cu113
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_wheel_py3.9_cu113_upload
          requires:
          - nightly_binary_linux_wheel_py3.9_cu113
          subfolder: cu113/
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_wheel_py3.9_cu113_smoke_test_pip
          python_version: '3.9'
          requires:
          - nightly_binary_linux_wheel_py3.9_cu113_upload
Jeff Daily's avatar
Jeff Daily committed
2398
      - binary_linux_wheel:
2399
          cu_version: rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2400
2401
2402
2403
2404
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2405
          name: nightly_binary_linux_wheel_py3.9_rocm4.2
Jeff Daily's avatar
Jeff Daily committed
2406
          python_version: '3.9'
2407
          wheel_docker_image: pytorch/manylinux-rocm:4.2
Jeff Daily's avatar
Jeff Daily committed
2408
2409
2410
2411
2412
2413
2414
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2415
          name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
Jeff Daily's avatar
Jeff Daily committed
2416
          requires:
2417
2418
          - nightly_binary_linux_wheel_py3.9_rocm4.2
          subfolder: rocm4.2/
Jeff Daily's avatar
Jeff Daily committed
2419
2420
2421
2422
2423
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2424
          name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
2425
2426
          python_version: '3.9'
          requires:
2427
          - nightly_binary_linux_wheel_py3.9_rocm4.2_upload
2428
      - binary_linux_wheel:
2429
          cu_version: rocm4.3.1
2430
2431
2432
2433
2434
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2435
          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
2436
          python_version: '3.9'
2437
          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
2438
2439
2440
2441
2442
2443
2444
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2445
          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
2446
          requires:
2447
2448
          - nightly_binary_linux_wheel_py3.9_rocm4.3.1
          subfolder: rocm4.3.1/
2449
2450
2451
2452
2453
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2454
          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
2455
2456
          python_version: '3.9'
          requires:
2457
          - nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
2458
2459
2460
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
2461
2462
2463
          filters:
            branches:
              only: nightly
2464
2465
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2466
          name: nightly_binary_macos_wheel_py3.6_cpu
2467
          python_version: '3.6'
2468
          wheel_docker_image: pytorch/manylinux-cuda102
2469
2470
2471
2472
2473
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2474
2475
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2476
          name: nightly_binary_macos_wheel_py3.6_cpu_upload
2477
          requires:
2478
2479
2480
2481
2482
2483
2484
2485
2486
2487
2488
2489
2490
2491
2492
2493
2494
2495
2496
2497
2498
2499
2500
2501
2502
2503
2504
2505
2506
2507
2508
2509
2510
2511
2512
2513
2514
2515
2516
2517
2518
2519
2520
2521
2522
2523
2524
2525
2526
2527
2528
2529
2530
2531
2532
2533
2534
2535
2536
2537
2538
2539
2540
2541
2542
2543
2544
2545
2546
2547
2548
2549
2550
2551
2552
2553
2554
2555
2556
2557
2558
2559
2560
2561
2562
2563
2564
2565
2566
2567
2568
2569
2570
2571
2572
2573
2574
          - nightly_binary_macos_wheel_py3.6_cpu
          subfolder: ''
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.7_cpu
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.7_cpu_upload
          requires:
          - nightly_binary_macos_wheel_py3.7_cpu
          subfolder: ''
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.8_cpu
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.8_cpu_upload
          requires:
          - nightly_binary_macos_wheel_py3.8_cpu
          subfolder: ''
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_wheel_py3.9_cpu_upload
          requires:
          - nightly_binary_macos_wheel_py3.9_cpu
          subfolder: ''
      - binary_win_wheel:
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.6_cpu
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.6_cpu_upload
          requires:
          - nightly_binary_win_wheel_py3.6_cpu
          subfolder: cpu/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.6_cpu_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_win_wheel_py3.6_cpu_upload
peterjc123's avatar
peterjc123 committed
2575
      - binary_win_wheel:
2576
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2577
2578
2579
2580
2581
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2582
          name: nightly_binary_win_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2583
2584
2585
2586
2587
2588
2589
2590
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2591
          name: nightly_binary_win_wheel_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
2592
          requires:
2593
2594
          - nightly_binary_win_wheel_py3.6_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2595
2596
2597
2598
2599
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2600
          name: nightly_binary_win_wheel_py3.6_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2601
2602
          python_version: '3.6'
          requires:
2603
          - nightly_binary_win_wheel_py3.6_cu111_upload
2604
2605
2606
2607
2608
2609
2610
2611
2612
2613
2614
2615
2616
2617
2618
2619
2620
2621
2622
2623
2624
2625
2626
2627
2628
2629
2630
2631
2632
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.6_cu113
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.6_cu113_upload
          requires:
          - nightly_binary_win_wheel_py3.6_cu113
          subfolder: cu113/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.6_cu113_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_win_wheel_py3.6_cu113_upload
2633
      - binary_win_wheel:
2634
2635
2636
2637
          cu_version: cpu
          filters:
            branches:
              only: nightly
2638
2639
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2640
2641
2642
2643
2644
2645
2646
          name: nightly_binary_win_wheel_py3.7_cpu
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2647
2648
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2649
2650
2651
2652
          name: nightly_binary_win_wheel_py3.7_cpu_upload
          requires:
          - nightly_binary_win_wheel_py3.7_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
2653
2654
2655
2656
2657
2658
2659
2660
2661
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.7_cpu_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_win_wheel_py3.7_cpu_upload
peterjc123's avatar
peterjc123 committed
2662
      - binary_win_wheel:
2663
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2664
2665
2666
2667
2668
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2669
          name: nightly_binary_win_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
2670
2671
2672
2673
2674
2675
2676
2677
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2678
          name: nightly_binary_win_wheel_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
2679
          requires:
2680
2681
          - nightly_binary_win_wheel_py3.7_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2682
2683
2684
2685
2686
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2687
          name: nightly_binary_win_wheel_py3.7_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2688
2689
          python_version: '3.7'
          requires:
2690
          - nightly_binary_win_wheel_py3.7_cu111_upload
2691
2692
2693
2694
2695
2696
2697
2698
2699
2700
2701
2702
2703
2704
2705
2706
2707
2708
2709
2710
2711
2712
2713
2714
2715
2716
2717
2718
2719
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.7_cu113
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.7_cu113_upload
          requires:
          - nightly_binary_win_wheel_py3.7_cu113
          subfolder: cu113/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.7_cu113_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_win_wheel_py3.7_cu113_upload
2720
      - binary_win_wheel:
2721
2722
2723
2724
          cu_version: cpu
          filters:
            branches:
              only: nightly
2725
2726
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2727
2728
2729
2730
2731
2732
2733
          name: nightly_binary_win_wheel_py3.8_cpu
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2734
2735
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2736
2737
2738
2739
          name: nightly_binary_win_wheel_py3.8_cpu_upload
          requires:
          - nightly_binary_win_wheel_py3.8_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
2740
2741
2742
2743
2744
2745
2746
2747
2748
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.8_cpu_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_win_wheel_py3.8_cpu_upload
peterjc123's avatar
peterjc123 committed
2749
      - binary_win_wheel:
2750
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2751
2752
2753
2754
2755
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2756
          name: nightly_binary_win_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
2757
2758
2759
2760
2761
2762
2763
2764
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2765
          name: nightly_binary_win_wheel_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
2766
          requires:
2767
2768
          - nightly_binary_win_wheel_py3.8_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2769
2770
2771
2772
2773
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2774
          name: nightly_binary_win_wheel_py3.8_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2775
2776
          python_version: '3.8'
          requires:
2777
          - nightly_binary_win_wheel_py3.8_cu111_upload
2778
2779
2780
2781
2782
2783
2784
2785
2786
2787
2788
2789
2790
2791
2792
2793
2794
2795
2796
2797
2798
2799
2800
2801
2802
2803
2804
2805
2806
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.8_cu113
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.8_cu113_upload
          requires:
          - nightly_binary_win_wheel_py3.8_cu113
          subfolder: cu113/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.8_cu113_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_win_wheel_py3.8_cu113_upload
2807
      - binary_win_wheel:
2808
          cu_version: cpu
2809
2810
2811
          filters:
            branches:
              only: nightly
2812
2813
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2814
2815
2816
          name: nightly_binary_win_wheel_py3.9_cpu
          python_version: '3.9'
      - binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
2817
          context: org-member
2818
2819
2820
          filters:
            branches:
              only: nightly
2821
2822
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2823
          name: nightly_binary_win_wheel_py3.9_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2824
          requires:
2825
2826
2827
          - nightly_binary_win_wheel_py3.9_cpu
          subfolder: cpu/
      - smoke_test_win_pip:
guyang3532's avatar
guyang3532 committed
2828
2829
2830
2831
          filters:
            branches:
              only:
              - nightly
2832
2833
          name: nightly_binary_win_wheel_py3.9_cpu_smoke_test_pip
          python_version: '3.9'
guyang3532's avatar
guyang3532 committed
2834
          requires:
2835
2836
          - nightly_binary_win_wheel_py3.9_cpu_upload
      - binary_win_wheel:
2837
          cu_version: cu111
2838
2839
2840
2841
2842
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2843
          name: nightly_binary_win_wheel_py3.9_cu111
2844
2845
2846
2847
2848
2849
2850
2851
          python_version: '3.9'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2852
          name: nightly_binary_win_wheel_py3.9_cu111_upload
2853
          requires:
2854
2855
          - nightly_binary_win_wheel_py3.9_cu111
          subfolder: cu111/
2856
2857
2858
2859
2860
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2861
          name: nightly_binary_win_wheel_py3.9_cu111_smoke_test_pip
2862
2863
          python_version: '3.9'
          requires:
2864
          - nightly_binary_win_wheel_py3.9_cu111_upload
2865
2866
2867
2868
2869
2870
2871
2872
2873
2874
2875
2876
2877
2878
2879
2880
2881
2882
2883
2884
2885
2886
2887
2888
2889
2890
2891
2892
2893
      - binary_win_wheel:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.9_cu113
          python_version: '3.9'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.9_cu113_upload
          requires:
          - nightly_binary_win_wheel_py3.9_cu113
          subfolder: cu113/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.9_cu113_smoke_test_pip
          python_version: '3.9'
          requires:
          - nightly_binary_win_wheel_py3.9_cu113_upload
2894
      - binary_linux_conda:
2895
2896
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
2897
2898
2899
          filters:
            branches:
              only: nightly
2900
2901
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2902
          name: nightly_binary_linux_conda_py3.6_cpu
2903
          python_version: '3.6'
2904
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
2905
2906
      - binary_conda_upload:
          context: org-member
2907
2908
2909
          filters:
            branches:
              only: nightly
2910
2911
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2912
          name: nightly_binary_linux_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2913
          requires:
2914
          - nightly_binary_linux_conda_py3.6_cpu
guyang3532's avatar
guyang3532 committed
2915
2916
2917
2918
2919
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2920
          name: nightly_binary_linux_conda_py3.6_cpu_smoke_test_conda
guyang3532's avatar
guyang3532 committed
2921
2922
          python_version: '3.6'
          requires:
2923
          - nightly_binary_linux_conda_py3.6_cpu_upload
Francisco Massa's avatar
Francisco Massa committed
2924
      - binary_linux_conda:
2925
          conda_docker_image: pytorch/conda-builder:cuda102
2926
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
2927
2928
2929
          filters:
            branches:
              only: nightly
2930
2931
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2932
          name: nightly_binary_linux_conda_py3.6_cu102
Francisco Massa's avatar
Francisco Massa committed
2933
          python_version: '3.6'
2934
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
2935
2936
2937
2938
2939
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2940
2941
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2942
          name: nightly_binary_linux_conda_py3.6_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
2943
          requires:
2944
          - nightly_binary_linux_conda_py3.6_cu102
guyang3532's avatar
guyang3532 committed
2945
2946
2947
2948
2949
2950
2951
2952
2953
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.6_cu102_smoke_test_conda
          python_version: '3.6'
          requires:
          - nightly_binary_linux_conda_py3.6_cu102_upload
peterjc123's avatar
peterjc123 committed
2954
      - binary_linux_conda:
2955
2956
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2957
2958
2959
2960
2961
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2962
          name: nightly_binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2963
          python_version: '3.6'
2964
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
2965
2966
2967
2968
2969
2970
2971
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2972
          name: nightly_binary_linux_conda_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
2973
          requires:
2974
          - nightly_binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2975
2976
2977
2978
2979
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2980
          name: nightly_binary_linux_conda_py3.6_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
2981
2982
          python_version: '3.6'
          requires:
2983
          - nightly_binary_linux_conda_py3.6_cu111_upload
2984
2985
2986
2987
2988
2989
2990
2991
2992
2993
2994
2995
2996
2997
2998
2999
3000
3001
3002
3003
3004
3005
3006
3007
3008
3009
3010
3011
3012
3013
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.6_cu113
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.6_cu113_upload
          requires:
          - nightly_binary_linux_conda_py3.6_cu113
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.6_cu113_smoke_test_conda
          python_version: '3.6'
          requires:
          - nightly_binary_linux_conda_py3.6_cu113_upload
3014
      - binary_linux_conda:
3015
          conda_docker_image: pytorch/conda-builder:cpu
3016
          cu_version: cpu
3017
3018
3019
          filters:
            branches:
              only: nightly
3020
3021
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
3022
          name: nightly_binary_linux_conda_py3.7_cpu
3023
          python_version: '3.7'
3024
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
3025
3026
      - binary_conda_upload:
          context: org-member
3027
3028
3029
          filters:
            branches:
              only: nightly
3030
3031
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3032
          name: nightly_binary_linux_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
3033
          requires:
3034
          - nightly_binary_linux_conda_py3.7_cpu
guyang3532's avatar
guyang3532 committed
3035
3036
3037
3038
3039
3040
3041
3042
3043
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.7_cpu_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_linux_conda_py3.7_cpu_upload
Francisco Massa's avatar
Francisco Massa committed
3044
      - binary_linux_conda:
3045
          conda_docker_image: pytorch/conda-builder:cuda102
3046
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
3047
3048
3049
          filters:
            branches:
              only: nightly
3050
3051
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3052
          name: nightly_binary_linux_conda_py3.7_cu102
Francisco Massa's avatar
Francisco Massa committed
3053
          python_version: '3.7'
3054
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
3055
3056
3057
3058
3059
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3060
3061
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3062
          name: nightly_binary_linux_conda_py3.7_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
3063
          requires:
3064
          - nightly_binary_linux_conda_py3.7_cu102
guyang3532's avatar
guyang3532 committed
3065
3066
3067
3068
3069
3070
3071
3072
3073
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.7_cu102_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_linux_conda_py3.7_cu102_upload
peterjc123's avatar
peterjc123 committed
3074
      - binary_linux_conda:
3075
3076
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3077
3078
3079
3080
3081
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3082
          name: nightly_binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3083
          python_version: '3.7'
3084
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
3085
3086
3087
3088
3089
3090
3091
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3092
          name: nightly_binary_linux_conda_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
3093
          requires:
3094
          - nightly_binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3095
3096
3097
3098
3099
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3100
          name: nightly_binary_linux_conda_py3.7_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3101
3102
          python_version: '3.7'
          requires:
3103
          - nightly_binary_linux_conda_py3.7_cu111_upload
3104
3105
3106
3107
3108
3109
3110
3111
3112
3113
3114
3115
3116
3117
3118
3119
3120
3121
3122
3123
3124
3125
3126
3127
3128
3129
3130
3131
3132
3133
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.7_cu113
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.7_cu113_upload
          requires:
          - nightly_binary_linux_conda_py3.7_cu113
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.7_cu113_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_linux_conda_py3.7_cu113_upload
3134
      - binary_linux_conda:
3135
          conda_docker_image: pytorch/conda-builder:cpu
3136
3137
3138
3139
          cu_version: cpu
          filters:
            branches:
              only: nightly
3140
3141
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3142
3143
          name: nightly_binary_linux_conda_py3.8_cpu
          python_version: '3.8'
3144
          wheel_docker_image: pytorch/manylinux-cuda102
3145
3146
3147
3148
3149
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3150
3151
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3152
3153
3154
          name: nightly_binary_linux_conda_py3.8_cpu_upload
          requires:
          - nightly_binary_linux_conda_py3.8_cpu
guyang3532's avatar
guyang3532 committed
3155
3156
3157
3158
3159
3160
3161
3162
3163
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.8_cpu_smoke_test_conda
          python_version: '3.8'
          requires:
          - nightly_binary_linux_conda_py3.8_cpu_upload
3164
      - binary_linux_conda:
3165
3166
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
3167
3168
3169
          filters:
            branches:
              only: nightly
3170
3171
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3172
          name: nightly_binary_linux_conda_py3.8_cu102
3173
          python_version: '3.8'
3174
          wheel_docker_image: pytorch/manylinux-cuda102
3175
3176
3177
3178
3179
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3180
3181
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3182
          name: nightly_binary_linux_conda_py3.8_cu102_upload
3183
          requires:
3184
          - nightly_binary_linux_conda_py3.8_cu102
guyang3532's avatar
guyang3532 committed
3185
3186
3187
3188
3189
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3190
          name: nightly_binary_linux_conda_py3.8_cu102_smoke_test_conda
guyang3532's avatar
guyang3532 committed
3191
3192
          python_version: '3.8'
          requires:
3193
          - nightly_binary_linux_conda_py3.8_cu102_upload
3194
      - binary_linux_conda:
3195
3196
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
3197
3198
3199
          filters:
            branches:
              only: nightly
3200
3201
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3202
          name: nightly_binary_linux_conda_py3.8_cu111
3203
          python_version: '3.8'
3204
          wheel_docker_image: pytorch/manylinux-cuda111
3205
3206
3207
3208
3209
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3210
3211
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3212
          name: nightly_binary_linux_conda_py3.8_cu111_upload
3213
          requires:
3214
          - nightly_binary_linux_conda_py3.8_cu111
guyang3532's avatar
guyang3532 committed
3215
3216
3217
3218
3219
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3220
          name: nightly_binary_linux_conda_py3.8_cu111_smoke_test_conda
guyang3532's avatar
guyang3532 committed
3221
3222
          python_version: '3.8'
          requires:
3223
          - nightly_binary_linux_conda_py3.8_cu111_upload
3224
3225
3226
3227
3228
3229
3230
3231
3232
3233
3234
3235
3236
3237
3238
3239
3240
3241
3242
3243
3244
3245
3246
3247
3248
3249
3250
3251
3252
3253
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.8_cu113
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.8_cu113_upload
          requires:
          - nightly_binary_linux_conda_py3.8_cu113
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.8_cu113_smoke_test_conda
          python_version: '3.8'
          requires:
          - nightly_binary_linux_conda_py3.8_cu113_upload
peterjc123's avatar
peterjc123 committed
3254
      - binary_linux_conda:
3255
3256
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
peterjc123's avatar
peterjc123 committed
3257
3258
3259
3260
3261
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3262
3263
3264
          name: nightly_binary_linux_conda_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
3265
3266
3267
3268
3269
3270
3271
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3272
          name: nightly_binary_linux_conda_py3.9_cpu_upload
peterjc123's avatar
peterjc123 committed
3273
          requires:
3274
          - nightly_binary_linux_conda_py3.9_cpu
peterjc123's avatar
peterjc123 committed
3275
3276
3277
3278
3279
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3280
3281
3282
3283
3284
3285
3286
3287
3288
3289
3290
3291
3292
3293
3294
3295
3296
3297
3298
3299
3300
3301
3302
3303
3304
3305
3306
3307
3308
3309
3310
3311
3312
3313
3314
          name: nightly_binary_linux_conda_py3.9_cpu_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_linux_conda_py3.9_cpu_upload
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.9_cu102
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.9_cu102_upload
          requires:
          - nightly_binary_linux_conda_py3.9_cu102
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.9_cu102_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_linux_conda_py3.9_cu102_upload
      - binary_linux_conda:
3315
3316
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
3317
3318
3319
3320
3321
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3322
          name: nightly_binary_linux_conda_py3.9_cu111
3323
          python_version: '3.9'
3324
          wheel_docker_image: pytorch/manylinux-cuda111
3325
3326
3327
3328
3329
3330
3331
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3332
          name: nightly_binary_linux_conda_py3.9_cu111_upload
peterjc123's avatar
peterjc123 committed
3333
          requires:
3334
          - nightly_binary_linux_conda_py3.9_cu111
3335
3336
3337
3338
3339
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3340
          name: nightly_binary_linux_conda_py3.9_cu111_smoke_test_conda
3341
3342
          python_version: '3.9'
          requires:
3343
          - nightly_binary_linux_conda_py3.9_cu111_upload
3344
3345
3346
3347
3348
3349
3350
3351
3352
3353
3354
3355
3356
3357
3358
3359
3360
3361
3362
3363
3364
3365
3366
3367
3368
3369
3370
3371
3372
3373
      - binary_linux_conda:
          conda_docker_image: pytorch/conda-builder:cuda113
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.9_cu113
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda113
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_linux_conda_py3.9_cu113_upload
          requires:
          - nightly_binary_linux_conda_py3.9_cu113
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_linux_conda_py3.9_cu113_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_linux_conda_py3.9_cu113_upload
3374
      - binary_macos_conda:
3375
          conda_docker_image: pytorch/conda-builder:cpu
3376
          cu_version: cpu
3377
3378
3379
          filters:
            branches:
              only: nightly
3380
3381
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
3382
          name: nightly_binary_macos_conda_py3.6_cpu
3383
          python_version: '3.6'
3384
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
3385
3386
      - binary_conda_upload:
          context: org-member
3387
3388
3389
          filters:
            branches:
              only: nightly
3390
3391
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3392
          name: nightly_binary_macos_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
3393
          requires:
3394
          - nightly_binary_macos_conda_py3.6_cpu
3395
      - binary_macos_conda:
3396
          conda_docker_image: pytorch/conda-builder:cpu
3397
          cu_version: cpu
3398
3399
3400
          filters:
            branches:
              only: nightly
3401
3402
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
3403
          name: nightly_binary_macos_conda_py3.7_cpu
3404
          python_version: '3.7'
3405
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
3406
3407
      - binary_conda_upload:
          context: org-member
3408
3409
3410
          filters:
            branches:
              only: nightly
3411
3412
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3413
          name: nightly_binary_macos_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
3414
          requires:
3415
3416
          - nightly_binary_macos_conda_py3.7_cpu
      - binary_macos_conda:
3417
          conda_docker_image: pytorch/conda-builder:cpu
3418
3419
3420
3421
          cu_version: cpu
          filters:
            branches:
              only: nightly
3422
3423
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3424
3425
          name: nightly_binary_macos_conda_py3.8_cpu
          python_version: '3.8'
3426
          wheel_docker_image: pytorch/manylinux-cuda102
3427
3428
3429
3430
3431
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3432
3433
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3434
3435
          name: nightly_binary_macos_conda_py3.8_cpu_upload
          requires:
3436
          - nightly_binary_macos_conda_py3.8_cpu
3437
3438
3439
3440
3441
3442
3443
3444
3445
3446
3447
3448
3449
3450
3451
3452
3453
3454
3455
3456
3457
      - binary_macos_conda:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_conda_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_macos_conda_py3.9_cpu_upload
          requires:
          - nightly_binary_macos_conda_py3.9_cpu
3458
      - binary_win_conda:
3459
3460
3461
3462
          cu_version: cpu
          filters:
            branches:
              only: nightly
3463
3464
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3465
3466
3467
3468
3469
3470
3471
          name: nightly_binary_win_conda_py3.6_cpu
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3472
3473
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3474
3475
3476
          name: nightly_binary_win_conda_py3.6_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.6_cpu
guyang3532's avatar
guyang3532 committed
3477
3478
3479
3480
3481
3482
3483
3484
3485
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.6_cpu_smoke_test_conda
          python_version: '3.6'
          requires:
          - nightly_binary_win_conda_py3.6_cpu_upload
peterjc123's avatar
peterjc123 committed
3486
      - binary_win_conda:
3487
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3488
3489
3490
3491
3492
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3493
          name: nightly_binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
3494
3495
3496
3497
3498
3499
3500
3501
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3502
          name: nightly_binary_win_conda_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
3503
          requires:
3504
          - nightly_binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
3505
3506
3507
3508
3509
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3510
          name: nightly_binary_win_conda_py3.6_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3511
3512
          python_version: '3.6'
          requires:
3513
          - nightly_binary_win_conda_py3.6_cu111_upload
3514
3515
3516
3517
3518
3519
3520
3521
3522
3523
3524
3525
3526
3527
3528
3529
3530
3531
3532
3533
3534
3535
3536
3537
3538
3539
3540
3541
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.6_cu113
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.6_cu113_upload
          requires:
          - nightly_binary_win_conda_py3.6_cu113
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.6_cu113_smoke_test_conda
          python_version: '3.6'
          requires:
          - nightly_binary_win_conda_py3.6_cu113_upload
3542
      - binary_win_conda:
3543
3544
3545
3546
          cu_version: cpu
          filters:
            branches:
              only: nightly
3547
3548
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3549
3550
3551
3552
3553
3554
3555
          name: nightly_binary_win_conda_py3.7_cpu
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3556
3557
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3558
3559
3560
          name: nightly_binary_win_conda_py3.7_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.7_cpu
guyang3532's avatar
guyang3532 committed
3561
3562
3563
3564
3565
3566
3567
3568
3569
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.7_cpu_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_win_conda_py3.7_cpu_upload
peterjc123's avatar
peterjc123 committed
3570
      - binary_win_conda:
3571
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3572
3573
3574
3575
3576
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3577
          name: nightly_binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3578
3579
3580
3581
3582
3583
3584
3585
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3586
          name: nightly_binary_win_conda_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
3587
          requires:
3588
          - nightly_binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3589
3590
3591
3592
3593
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3594
          name: nightly_binary_win_conda_py3.7_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3595
3596
          python_version: '3.7'
          requires:
3597
          - nightly_binary_win_conda_py3.7_cu111_upload
3598
3599
3600
3601
3602
3603
3604
3605
3606
3607
3608
3609
3610
3611
3612
3613
3614
3615
3616
3617
3618
3619
3620
3621
3622
3623
3624
3625
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.7_cu113
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.7_cu113_upload
          requires:
          - nightly_binary_win_conda_py3.7_cu113
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.7_cu113_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_win_conda_py3.7_cu113_upload
3626
      - binary_win_conda:
3627
3628
3629
3630
          cu_version: cpu
          filters:
            branches:
              only: nightly
3631
3632
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3633
3634
3635
3636
3637
3638
3639
          name: nightly_binary_win_conda_py3.8_cpu
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3640
3641
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3642
3643
3644
          name: nightly_binary_win_conda_py3.8_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.8_cpu
guyang3532's avatar
guyang3532 committed
3645
3646
3647
3648
3649
3650
3651
3652
3653
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.8_cpu_smoke_test_conda
          python_version: '3.8'
          requires:
          - nightly_binary_win_conda_py3.8_cpu_upload
peterjc123's avatar
peterjc123 committed
3654
      - binary_win_conda:
3655
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3656
3657
3658
3659
3660
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3661
          name: nightly_binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
3662
3663
3664
3665
3666
3667
3668
3669
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3670
          name: nightly_binary_win_conda_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
3671
          requires:
3672
          - nightly_binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
3673
3674
3675
3676
3677
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3678
          name: nightly_binary_win_conda_py3.8_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3679
3680
          python_version: '3.8'
          requires:
3681
          - nightly_binary_win_conda_py3.8_cu111_upload
3682
3683
3684
3685
3686
3687
3688
3689
3690
3691
3692
3693
3694
3695
3696
3697
3698
3699
3700
3701
3702
3703
3704
3705
3706
3707
3708
3709
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.8_cu113
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.8_cu113_upload
          requires:
          - nightly_binary_win_conda_py3.8_cu113
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.8_cu113_smoke_test_conda
          python_version: '3.8'
          requires:
          - nightly_binary_win_conda_py3.8_cu113_upload
3710
3711
3712
3713
3714
3715
3716
3717
3718
3719
3720
3721
3722
3723
3724
3725
3726
3727
3728
3729
3730
3731
3732
3733
3734
3735
3736
3737
3738
      - binary_win_conda:
          cu_version: cpu
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.9_cpu
          python_version: '3.9'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.9_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.9_cpu
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.9_cpu_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_win_conda_py3.9_cpu_upload
      - binary_win_conda:
3739
          cu_version: cu111
3740
3741
3742
3743
3744
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3745
          name: nightly_binary_win_conda_py3.9_cu111
3746
3747
3748
3749
3750
3751
3752
3753
          python_version: '3.9'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3754
          name: nightly_binary_win_conda_py3.9_cu111_upload
3755
          requires:
3756
          - nightly_binary_win_conda_py3.9_cu111
3757
3758
3759
3760
3761
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3762
          name: nightly_binary_win_conda_py3.9_cu111_smoke_test_conda
3763
3764
          python_version: '3.9'
          requires:
3765
          - nightly_binary_win_conda_py3.9_cu111_upload
3766
3767
3768
3769
3770
3771
3772
3773
3774
3775
3776
3777
3778
3779
3780
3781
3782
3783
3784
3785
3786
3787
3788
3789
3790
3791
3792
3793
      - binary_win_conda:
          cu_version: cu113
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.9_cu113
          python_version: '3.9'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.9_cu113_upload
          requires:
          - nightly_binary_win_conda_py3.9_cu113
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.9_cu113_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_win_conda_py3.9_cu113_upload
guyang3532's avatar
guyang3532 committed
3794
3795
3796
3797
3798
3799
3800
  docker_build:
    triggers:
      - schedule:
          cron: "0 10 * * 0"
          filters:
            branches:
              only:
3801
                - main
guyang3532's avatar
guyang3532 committed
3802
3803
    jobs:
      - smoke_test_docker_image_build:
3804
          context: org-member