config.yml 112 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
32
33
34
35
#     - run:
#         name: Checkout merge branch
#         command: |
#           set -ex
#           BRANCH=$(git rev-parse --abbrev-ref HEAD)
#           if [[ "$BRANCH" != "master" ]]; then
#             git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
#             git checkout "merged/$CIRCLE_BRANCH"
#           fi
36
37
38
39
40
41
  designate_upload_channel:
    description: "inserts the correct upload channel into ${BASH_ENV}"
    steps:
      - run:
          name: adding UPLOAD_CHANNEL to BASH_ENV
          command: |
42
            our_upload_channel=test
43
            echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
44
45
46
47
48
49
50
51
  install_cuda_compatible_cmath:
    description: "Install CUDA compatible cmath"
    steps:
      - run:
          name: _HACK_ Install CUDA compatible cmath
          no_output_timeout: 1m
          command: |
              powershell .circleci/scripts/vs_install_cmath.ps1
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
102
103
104
105
  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

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

143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
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 >>

159
160
161
162
163
164
165
166
torchvision_android_params: &torchvision_android_params
  parameters:
    build_environment:
      type: string
      default: ""
  environment:
    BUILD_ENVIRONMENT: << parameters.build_environment >>

guyang3532's avatar
guyang3532 committed
167
168
169
170
171
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
    - image: torchvision/smoke_test:latest

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

184
185
186
187
188
189
190
191
  python_lint:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
            pip install --user --progress-bar off flake8 typing
192
193
194
195
196
197
198
199
200
            flake8 --config=setup.cfg .

  python_type_check:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
201
202
            sudo apt-get update -y
            sudo apt install -y libturbojpeg-dev
203
            pip install --user --progress-bar off mypy
204
            pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
205
            pip install --user --progress-bar off --editable .
206
            mypy --config-file mypy.ini
207

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

218
219
220
221
222
223
224
  clang_format:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
225
226
227
            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
228
            ./.circleci/unittest/linux/scripts/run-clang-format.py -r torchvision/csrc --clang-format-executable /opt/clang-format
229

Francisco Massa's avatar
Francisco Massa committed
230
231
232
233
234
235
236
237
238
239
240
241
  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
            pip install --user --progress-bar off --editable .
            python test/test_hub.py

242
243
244
245
246
247
248
249
250
251
252
  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
            pip install --user --progress-bar off --editable .
            pip install --user onnx
253
            pip install --user onnxruntime
254
255
            python test/test_onnx.py

256
257
258
  binary_linux_wheel:
    <<: *binary_common
    docker:
Edward Z. Yang's avatar
Edward Z. Yang committed
259
      - image: << parameters.wheel_docker_image >>
260
261
    resource_class: 2xlarge+
    steps:
262
      - checkout_merge
263
      - designate_upload_channel
264
265
266
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
267
268
269
270
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
271
272
273
274

  binary_linux_conda:
    <<: *binary_common
    docker:
275
      - image: "<< parameters.conda_docker_image >>"
276
    resource_class: 2xlarge+
277
    steps:
278
      - checkout_merge
279
      - designate_upload_channel
280
281
282
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
283
284
285
286
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"
287
288
      - store_test_results:
          path: build_results/
289

Francisco Massa's avatar
Francisco Massa committed
290
  binary_win_conda:
291
292
    <<: *binary_common
    executor: windows-cpu
293
294
    steps:
      - checkout_merge
295
      - designate_upload_channel
296
      - install_cuda_compatible_cmath
297
298
      - run:
          name: Build conda packages
299
          no_output_timeout: 20m
300
          command: |
301
302
303
304
305
306
307
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
            packaging/build_conda.sh
308
            rm /C/tools/miniconda3/conda-bld/win-64/vs${VC_YEAR}*.tar.bz2
309
      - store_artifacts:
310
          path: C:/tools/miniconda3/conda-bld/win-64
311
      - persist_to_workspace:
312
          root: C:/tools/miniconda3/conda-bld/win-64
313
314
315
316
317
          paths:
            - "*"
      - store_test_results:
          path: build_results/

318
  binary_win_wheel:
319
320
    <<: *binary_common
    executor: windows-cpu
321
322
    steps:
      - checkout_merge
323
      - designate_upload_channel
324
      - install_cuda_compatible_cmath
325
326
327
      - run:
          name: Build wheel packages
          command: |
328
329
330
331
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_wheel.sh
332
      - store_artifacts:
333
          path: dist
334
      - persist_to_workspace:
335
          root: dist
336
337
338
339
340
          paths:
            - "*"
      - store_test_results:
          path: build_results/

341
342
343
  binary_macos_wheel:
    <<: *binary_common
    macos:
344
      xcode: "12.0"
345
    steps:
346
      - checkout_merge
347
      - designate_upload_channel
348
349
350
351
352
353
354
355
356
357
358
      - 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
359
360
361
362
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
363

364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
  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"

401
402
403
404
405
406
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
  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"

437
438
439
  binary_macos_conda:
    <<: *binary_common
    macos:
440
      xcode: "12.0"
441
    steps:
442
      - checkout_merge
443
      - designate_upload_channel
444
445
446
447
448
449
450
451
452
      - 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
453
454
455
456
      - persist_to_workspace:
          root: /Users/distiller/miniconda3/conda-bld/osx-64
          paths:
            - "*"
457
458
      - store_test_results:
          path: build_results/
Edward Z. Yang's avatar
Edward Z. Yang committed
459
460
461
462
463
464
465
466

  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
467
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
468
469
470
471
472
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
473
            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
474
475
476

  # Requires org-member context
  binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
477
478
479
480
    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
481
482
483
484
485
    docker:
      - image: circleci/python:3.7
    steps:
      - attach_workspace:
          at: ~/workspace
486
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
487
488
489
490
491
492
493
494
495
496
      - 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
497
            for pkg in ~/workspace/*.whl; do
498
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
499
            done
500

guyang3532's avatar
guyang3532 committed
501
502
503
504
505
  smoke_test_linux_conda:
    <<: *smoke_test_common
    steps:
      - attach_workspace:
          at: ~/workspace
506
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
      - 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
525
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
      - 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
546
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
      - 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
566
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
567
568
569
570
571
572
      - 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
573
574
575
576
577
            CONDA_CHANNEL_FLAGS=""
            if [[ "${PYTHON_VERSION}" = 3.9 ]]; then
              CONDA_CHANNEL_FLAGS="-c=conda-forge"
            fi
            conda create ${CONDA_CHANNEL_FLAGS} -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
guyang3532's avatar
guyang3532 committed
578
            conda activate python${PYTHON_VERSION}
579
            conda install Pillow>=5.3.0
guyang3532's avatar
guyang3532 committed
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
            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
596
      - designate_upload_channel
guyang3532's avatar
guyang3532 committed
597
598
599
600
601
      - run:
          name: install binaries
          command: |
            set -x
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
602
603
604
605
606
            CONDA_CHANNEL_FLAGS=""
            if [[ "${PYTHON_VERSION}" = 3.9 ]]; then
              CONDA_CHANNEL_FLAGS="-c=conda-forge"
            fi
            conda create ${CONDA_CHANNEL_FLAGS} -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
guyang3532's avatar
guyang3532 committed
607
608
609
610
611
612
613
614
615
616
            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"

617
618
619
620
621
622
623
  unittest_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout
624
      - designate_upload_channel
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
      - 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:
659
      image: ubuntu-1604-cuda-10.2:202012-01
660
    resource_class: gpu.nvidia.medium
661
    environment:
662
      image_name: "pytorch/manylinux-cuda102"
663
      PYTHON_VERSION: << parameters.python_version >>
664
665
    steps:
      - checkout
666
      - designate_upload_channel
667
668
669
670
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
671
672
673
674
675
      - restore_cache:

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

676
677
      - run:
          name: Setup
678
          command: docker run -e PYTHON_VERSION -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/setup_env.sh
679
680
      - save_cache:

681
          key: env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".circleci-weekly" }}
682
683
684
685
686
687

          paths:
            - conda
            - env
      - run:
          name: Install torchvision
688
          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
689
690
      - run:
          name: Run tests
691
          command: docker run -e CIRCLECI -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
692
693
694
695
696
697
698
699
700
701
702
703
      - 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
704
      - designate_upload_channel
705
      - install_cuda_compatible_cmath
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
      - 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:
742
      CUDA_VERSION: "10.2"
743
      PYTHON_VERSION: << parameters.python_version >>
744
745
    steps:
      - checkout
746
      - designate_upload_channel
747
      - install_cuda_compatible_cmath
748
749
750
751
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
752
753
754
755
      - restore_cache:

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

757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
      - 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
      - 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
778

Francisco Massa's avatar
Francisco Massa committed
779
780
781
  unittest_macos_cpu:
    <<: *binary_common
    macos:
782
      xcode: "12.0"
Francisco Massa's avatar
Francisco Massa committed
783
784
785
    resource_class: large
    steps:
      - checkout
786
      - designate_upload_channel
Francisco Massa's avatar
Francisco Massa committed
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
      - 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

822
823
824
825
826
827
828
  cmake_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout_merge
829
      - designate_upload_channel
830
831
832
833
834
835
836
837
      - run:
          name: Setup conda
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run: packaging/build_cmake.sh

  cmake_linux_gpu:
    <<: *binary_common
    machine:
838
      image: ubuntu-1604-cuda-10.2:202012-01
839
840
841
842
843
844
845
846
    resource_class: gpu.small
    environment:
      PYTHON_VERSION: << parameters.python_version >>
      PYTORCH_VERSION: << parameters.pytorch_version >>
      UNICODE_ABI: << parameters.unicode_abi >>
      CU_VERSION: << parameters.cu_version >>
    steps:
      - checkout_merge
847
      - designate_upload_channel
848
849
850
851
852
      - 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
853
          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
854
855
856
857

  cmake_macos_cpu:
    <<: *binary_common
    macos:
858
      xcode: "12.0"
859
860
    steps:
      - checkout_merge
861
      - designate_upload_channel
862
863
864
865
866
867
868
869
870
871
872
873
874
875
      - 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
876
      - designate_upload_channel
877
      - install_cuda_compatible_cmath
878
879
880
881
882
883
884
885
886
887
888
889
      - 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
890
      - designate_upload_channel
891
      - install_cuda_compatible_cmath
892
893
894
895
896
897
898
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_cmake.sh

899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
  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
915
916
917
918
919
920
921
922
923
      - 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:
            - sphinx-gallery-{{ checksum "./docs/source/conf.py" }}-{{ checksum ".circleci-weekly" }}

924
925
926
927
928
929
930
931
932
933
934
935
      - run:
          name: Build docs
          command: |
            set -ex
            tag=${CIRCLE_TAG:1:5}
            VERSION=${tag:-master}
            eval "$(./conda/bin/conda shell.bash hook)"
            conda activate ./env
            pushd docs
            pip install -r requirements.txt
            make html
            popd
936
937
938
939
940
941
      - save_cache:

          key: sphinx-gallery-{{ checksum "./docs/source/conf.py" }}-{{ checksum ".circleci-weekly" }}

          paths:
            - ./docs/source/auto_examples
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
      - 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}
            target=${tag:-master}
            ~/workspace/.circleci/build_docs/commit_docs.sh ~/workspace $target


979
980
981
982
workflows:
  build:
    jobs:
      - circleci_consistency
Edward Z. Yang's avatar
Edward Z. Yang committed
983
      - binary_linux_wheel:
984
          conda_docker_image: pytorch/conda-builder:cpu
985
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
986
          name: binary_linux_wheel_py3.6_cpu
987
          python_version: '3.6'
988
989
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
990
          conda_docker_image: pytorch/conda-builder:cuda102
991
992
993
994
          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
995
      - binary_linux_wheel:
996
997
998
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
999
          python_version: '3.6'
1000
          wheel_docker_image: pytorch/manylinux-cuda111
1001
1002
1003
1004
1005
      - binary_linux_wheel:
          cu_version: rocm4.1
          name: binary_linux_wheel_py3.6_rocm4.1
          python_version: '3.6'
          wheel_docker_image: pytorch/manylinux-rocm:4.1
1006
1007
1008
1009
1010
      - 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
Edward Z. Yang's avatar
Edward Z. Yang committed
1011
      - binary_linux_wheel:
1012
          conda_docker_image: pytorch/conda-builder:cpu
1013
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1014
          name: binary_linux_wheel_py3.7_cpu
1015
          python_version: '3.7'
1016
1017
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1018
          conda_docker_image: pytorch/conda-builder:cuda102
1019
1020
1021
1022
          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
1023
      - binary_linux_wheel:
1024
1025
1026
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1027
          python_version: '3.7'
1028
          wheel_docker_image: pytorch/manylinux-cuda111
1029
1030
1031
1032
1033
      - binary_linux_wheel:
          cu_version: rocm4.1
          name: binary_linux_wheel_py3.7_rocm4.1
          python_version: '3.7'
          wheel_docker_image: pytorch/manylinux-rocm:4.1
1034
1035
1036
1037
1038
      - 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
1039
      - binary_linux_wheel:
1040
          conda_docker_image: pytorch/conda-builder:cpu
1041
1042
1043
          cu_version: cpu
          name: binary_linux_wheel_py3.8_cpu
          python_version: '3.8'
1044
1045
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_wheel:
1046
          conda_docker_image: pytorch/conda-builder:cuda102
1047
1048
1049
1050
          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
1051
      - binary_linux_wheel:
1052
1053
1054
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1055
          python_version: '3.8'
1056
          wheel_docker_image: pytorch/manylinux-cuda111
1057
1058
1059
1060
1061
      - binary_linux_wheel:
          cu_version: rocm4.1
          name: binary_linux_wheel_py3.8_rocm4.1
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-rocm:4.1
1062
1063
1064
1065
1066
      - 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
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
      - 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:
1080
1081
1082
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_wheel_py3.9_cu111
1083
          python_version: '3.9'
1084
          wheel_docker_image: pytorch/manylinux-cuda111
1085
1086
1087
1088
1089
      - binary_linux_wheel:
          cu_version: rocm4.1
          name: binary_linux_wheel_py3.9_rocm4.1
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-rocm:4.1
1090
1091
1092
1093
1094
      - 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
Edward Z. Yang's avatar
Edward Z. Yang committed
1095
      - binary_macos_wheel:
1096
          conda_docker_image: pytorch/conda-builder:cpu
1097
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1098
          name: binary_macos_wheel_py3.6_cpu
1099
          python_version: '3.6'
1100
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1101
      - binary_macos_wheel:
1102
          conda_docker_image: pytorch/conda-builder:cpu
1103
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1104
          name: binary_macos_wheel_py3.7_cpu
1105
          python_version: '3.7'
1106
          wheel_docker_image: pytorch/manylinux-cuda102
1107
      - binary_macos_wheel:
1108
          conda_docker_image: pytorch/conda-builder:cpu
1109
1110
1111
          cu_version: cpu
          name: binary_macos_wheel_py3.8_cpu
          python_version: '3.8'
1112
          wheel_docker_image: pytorch/manylinux-cuda102
1113
1114
1115
1116
1117
1118
      - 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
1119
      - binary_win_wheel:
1120
1121
1122
1123
          cu_version: cpu
          filters:
            branches:
              only: master
1124
1125
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1126
1127
          name: binary_win_wheel_py3.6_cpu
          python_version: '3.6'
1128
      - binary_win_wheel:
1129
          cu_version: cu102
1130
1131
1132
          filters:
            branches:
              only: master
1133
1134
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1135
1136
          name: binary_win_wheel_py3.6_cu102
          python_version: '3.6'
peterjc123's avatar
peterjc123 committed
1137
      - binary_win_wheel:
1138
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1139
1140
1141
1142
1143
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1144
          name: binary_win_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1145
          python_version: '3.6'
1146
      - binary_win_wheel:
1147
1148
1149
1150
          cu_version: cpu
          filters:
            branches:
              only: master
1151
1152
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1153
1154
          name: binary_win_wheel_py3.7_cpu
          python_version: '3.7'
1155
      - binary_win_wheel:
1156
          cu_version: cu102
1157
1158
1159
          filters:
            branches:
              only: master
1160
1161
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1162
1163
          name: binary_win_wheel_py3.7_cu102
          python_version: '3.7'
peterjc123's avatar
peterjc123 committed
1164
      - binary_win_wheel:
1165
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1166
1167
1168
1169
1170
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1171
          name: binary_win_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1172
          python_version: '3.7'
1173
      - binary_win_wheel:
1174
          cu_version: cpu
1175
1176
1177
1178
1179
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1180
1181
          name: binary_win_wheel_py3.8_cpu
          python_version: '3.8'
1182
      - binary_win_wheel:
1183
          cu_version: cu102
peterjc123's avatar
peterjc123 committed
1184
1185
1186
1187
1188
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1189
1190
          name: binary_win_wheel_py3.8_cu102
          python_version: '3.8'
peterjc123's avatar
peterjc123 committed
1191
      - binary_win_wheel:
1192
          cu_version: cu111
1193
1194
1195
1196
1197
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1198
          name: binary_win_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1199
          python_version: '3.8'
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
      - binary_win_wheel:
          cu_version: cpu
          name: binary_win_wheel_py3.9_cpu
          python_version: '3.9'
      - binary_win_wheel:
          cu_version: cu102
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_wheel_py3.9_cu102
          python_version: '3.9'
      - binary_win_wheel:
1214
1215
          cu_version: cu111
          name: binary_win_wheel_py3.9_cu111
1216
          python_version: '3.9'
Edward Z. Yang's avatar
Edward Z. Yang committed
1217
      - binary_linux_conda:
1218
          conda_docker_image: pytorch/conda-builder:cpu
1219
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1220
          name: binary_linux_conda_py3.6_cpu
1221
          python_version: '3.6'
1222
1223
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1224
          conda_docker_image: pytorch/conda-builder:cuda102
1225
1226
1227
1228
          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
1229
      - binary_linux_conda:
1230
1231
1232
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1233
          python_version: '3.6'
1234
          wheel_docker_image: pytorch/manylinux-cuda111
Edward Z. Yang's avatar
Edward Z. Yang committed
1235
      - binary_linux_conda:
1236
          conda_docker_image: pytorch/conda-builder:cpu
1237
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1238
          name: binary_linux_conda_py3.7_cpu
1239
          python_version: '3.7'
1240
1241
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1242
          conda_docker_image: pytorch/conda-builder:cuda102
1243
1244
1245
1246
          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
1247
      - binary_linux_conda:
1248
1249
1250
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1251
          python_version: '3.7'
1252
          wheel_docker_image: pytorch/manylinux-cuda111
1253
      - binary_linux_conda:
1254
          conda_docker_image: pytorch/conda-builder:cpu
1255
1256
1257
          cu_version: cpu
          name: binary_linux_conda_py3.8_cpu
          python_version: '3.8'
1258
1259
          wheel_docker_image: pytorch/manylinux-cuda102
      - binary_linux_conda:
1260
          conda_docker_image: pytorch/conda-builder:cuda102
1261
1262
1263
1264
          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
1265
      - binary_linux_conda:
1266
1267
1268
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1269
          python_version: '3.8'
1270
          wheel_docker_image: pytorch/manylinux-cuda111
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
      - 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:
1284
1285
1286
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
          name: binary_linux_conda_py3.9_cu111
1287
          python_version: '3.9'
1288
          wheel_docker_image: pytorch/manylinux-cuda111
Edward Z. Yang's avatar
Edward Z. Yang committed
1289
      - binary_macos_conda:
1290
          conda_docker_image: pytorch/conda-builder:cpu
1291
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1292
          name: binary_macos_conda_py3.6_cpu
1293
          python_version: '3.6'
1294
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1295
      - binary_macos_conda:
1296
          conda_docker_image: pytorch/conda-builder:cpu
1297
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
1298
          name: binary_macos_conda_py3.7_cpu
1299
          python_version: '3.7'
1300
          wheel_docker_image: pytorch/manylinux-cuda102
1301
      - binary_macos_conda:
1302
          conda_docker_image: pytorch/conda-builder:cpu
1303
1304
1305
          cu_version: cpu
          name: binary_macos_conda_py3.8_cpu
          python_version: '3.8'
1306
          wheel_docker_image: pytorch/manylinux-cuda102
1307
1308
1309
1310
1311
1312
      - 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
1313
      - binary_win_conda:
1314
1315
1316
1317
          cu_version: cpu
          filters:
            branches:
              only: master
1318
1319
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1320
1321
          name: binary_win_conda_py3.6_cpu
          python_version: '3.6'
1322
      - binary_win_conda:
1323
          cu_version: cu102
1324
1325
1326
          filters:
            branches:
              only: master
1327
1328
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1329
1330
          name: binary_win_conda_py3.6_cu102
          python_version: '3.6'
peterjc123's avatar
peterjc123 committed
1331
      - binary_win_conda:
1332
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1333
1334
1335
1336
1337
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1338
          name: binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1339
          python_version: '3.6'
1340
      - binary_win_conda:
1341
1342
1343
1344
          cu_version: cpu
          filters:
            branches:
              only: master
1345
1346
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1347
1348
          name: binary_win_conda_py3.7_cpu
          python_version: '3.7'
1349
      - binary_win_conda:
1350
          cu_version: cu102
1351
1352
1353
          filters:
            branches:
              only: master
1354
1355
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1356
1357
          name: binary_win_conda_py3.7_cu102
          python_version: '3.7'
peterjc123's avatar
peterjc123 committed
1358
      - binary_win_conda:
1359
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1360
1361
1362
1363
1364
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1365
          name: binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
1366
          python_version: '3.7'
1367
      - binary_win_conda:
1368
          cu_version: cpu
1369
1370
1371
1372
1373
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1374
1375
          name: binary_win_conda_py3.8_cpu
          python_version: '3.8'
1376
      - binary_win_conda:
1377
          cu_version: cu102
peterjc123's avatar
peterjc123 committed
1378
1379
1380
1381
1382
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1383
1384
          name: binary_win_conda_py3.8_cu102
          python_version: '3.8'
peterjc123's avatar
peterjc123 committed
1385
      - binary_win_conda:
1386
          cu_version: cu111
1387
1388
1389
1390
1391
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1392
          name: binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
1393
          python_version: '3.8'
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
      - binary_win_conda:
          cu_version: cpu
          name: binary_win_conda_py3.9_cpu
          python_version: '3.9'
      - binary_win_conda:
          cu_version: cu102
          filters:
            branches:
              only: master
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: binary_win_conda_py3.9_cu102
          python_version: '3.9'
      - binary_win_conda:
1408
1409
          cu_version: cu111
          name: binary_win_conda_py3.9_cu111
1410
          python_version: '3.9'
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
      - build_docs:
          name: build_docs
          python_version: '3.7'
          requires:
          - binary_linux_wheel_py3.7_cpu
      - upload_docs:
          context: org-member
          filters:
            branches:
              only:
              - nightly
1422
1423
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1424
1425
1426
1427
          name: upload_docs
          python_version: '3.7'
          requires:
          - build_docs
1428
      - python_lint
1429
      - python_type_check
1430
      - docstring_parameters_sync
1431
      - clang_format
Francisco Massa's avatar
Francisco Massa committed
1432
      - torchhub_test
1433
      - torch_onnx_test
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
      - 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
1444
1445
1446
      - binary_android_build:
          build_environment: binary-libtorchvision_ops-android
          name: binary_libtorchvision_ops_android
Edward Z. Yang's avatar
Edward Z. Yang committed
1447

1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461
  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'
1462
1463
1464
1465
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.9
          python_version: '3.9'
1466
      - unittest_linux_gpu:
1467
          cu_version: cu102
1468
1469
1470
1471
1472
1473
1474
1475
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_linux_gpu_py3.6
          python_version: '3.6'
      - unittest_linux_gpu:
1476
          cu_version: cu102
1477
1478
1479
1480
1481
1482
1483
1484
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_linux_gpu_py3.7
          python_version: '3.7'
      - unittest_linux_gpu:
1485
          cu_version: cu102
1486
1487
          name: unittest_linux_gpu_py3.8
          python_version: '3.8'
1488
      - unittest_linux_gpu:
1489
          cu_version: cu102
1490
1491
1492
1493
1494
1495
1496
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_linux_gpu_py3.9
          python_version: '3.9'
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
      - 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'
1509
1510
1511
1512
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.9
          python_version: '3.9'
1513
      - unittest_windows_gpu:
1514
          cu_version: cu102
1515
1516
1517
1518
1519
1520
1521
1522
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_windows_gpu_py3.6
          python_version: '3.6'
      - unittest_windows_gpu:
1523
          cu_version: cu102
1524
1525
1526
1527
1528
1529
1530
1531
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_windows_gpu_py3.7
          python_version: '3.7'
      - unittest_windows_gpu:
1532
          cu_version: cu102
1533
1534
          name: unittest_windows_gpu_py3.8
          python_version: '3.8'
1535
      - unittest_windows_gpu:
1536
          cu_version: cu102
1537
1538
1539
1540
1541
1542
1543
          filters:
            branches:
              only:
              - master
              - nightly
          name: unittest_windows_gpu_py3.9
          python_version: '3.9'
Francisco Massa's avatar
Francisco Massa committed
1544
1545
1546
1547
1548
1549
1550
1551
1552
1553
1554
1555
      - 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'
1556
1557
1558
1559
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.9
          python_version: '3.9'
1560
1561
1562
1563
1564
1565
1566
1567

  cmake:
    jobs:
      - cmake_linux_cpu:
          cu_version: cpu
          name: cmake_linux_cpu
          python_version: '3.8'
      - cmake_linux_gpu:
1568
          cu_version: cu102
1569
1570
          name: cmake_linux_gpu
          python_version: '3.8'
1571
          wheel_docker_image: pytorch/manylinux-cuda102
1572
1573
1574
1575
      - cmake_windows_cpu:
          cu_version: cpu
          name: cmake_windows_cpu
          python_version: '3.8'
1576
      - cmake_windows_gpu:
1577
          cu_version: cu102
1578
1579
          name: cmake_windows_gpu
          python_version: '3.8'
1580
1581
1582
1583
1584
      - cmake_macos_cpu:
          cu_version: cpu
          name: cmake_macos_cpu
          python_version: '3.8'

Edward Z. Yang's avatar
Edward Z. Yang committed
1585
1586
1587
  nightly:
    jobs:
      - circleci_consistency
1588
      - python_lint
1589
      - python_type_check
1590
      - docstring_parameters_sync
1591
      - clang_format
Francisco Massa's avatar
Francisco Massa committed
1592
      - torchhub_test
1593
      - torch_onnx_test
1594
1595
1596
1597
1598
1599
1600
1601
1602
1603
1604
1605
1606
1607
1608
1609
1610
1611
1612
1613
1614
1615
1616
1617
1618
1619
1620
1621
      - 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
1622
1623
1624
1625
1626
1627
1628
1629
      - 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
1630
      - binary_linux_wheel:
1631
          conda_docker_image: pytorch/conda-builder:cpu
1632
          cu_version: cpu
1633
1634
1635
          filters:
            branches:
              only: nightly
1636
1637
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
1638
          name: nightly_binary_linux_wheel_py3.6_cpu
1639
          python_version: '3.6'
1640
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1641
1642
      - binary_wheel_upload:
          context: org-member
1643
1644
1645
          filters:
            branches:
              only: nightly
1646
1647
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1648
          name: nightly_binary_linux_wheel_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
1649
          requires:
1650
1651
          - nightly_binary_linux_wheel_py3.6_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
1652
1653
1654
1655
1656
1657
1658
1659
1660
      - 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
1661
      - binary_linux_wheel:
1662
          conda_docker_image: pytorch/conda-builder:cuda102
1663
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
1664
1665
1666
          filters:
            branches:
              only: nightly
1667
1668
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1669
          name: nightly_binary_linux_wheel_py3.6_cu102
Francisco Massa's avatar
Francisco Massa committed
1670
          python_version: '3.6'
1671
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
1672
1673
1674
1675
1676
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1677
1678
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1679
          name: nightly_binary_linux_wheel_py3.6_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
1680
          requires:
1681
1682
          - nightly_binary_linux_wheel_py3.6_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
1683
1684
1685
1686
1687
1688
1689
1690
1691
      - 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
1692
      - binary_linux_wheel:
1693
1694
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
1695
1696
1697
1698
1699
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1700
          name: nightly_binary_linux_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
1701
          python_version: '3.6'
1702
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
1703
1704
1705
1706
1707
1708
1709
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1710
          name: nightly_binary_linux_wheel_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
1711
          requires:
1712
1713
          - nightly_binary_linux_wheel_py3.6_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
1714
1715
1716
1717
1718
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1719
          name: nightly_binary_linux_wheel_py3.6_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
1720
1721
          python_version: '3.6'
          requires:
1722
          - nightly_binary_linux_wheel_py3.6_cu111_upload
Jeff Daily's avatar
Jeff Daily committed
1723
      - binary_linux_wheel:
1724
          cu_version: rocm4.1
Jeff Daily's avatar
Jeff Daily committed
1725
1726
1727
1728
1729
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1730
          name: nightly_binary_linux_wheel_py3.6_rocm4.1
Jeff Daily's avatar
Jeff Daily committed
1731
          python_version: '3.6'
1732
          wheel_docker_image: pytorch/manylinux-rocm:4.1
Jeff Daily's avatar
Jeff Daily committed
1733
1734
1735
1736
1737
1738
1739
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1740
          name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload
Jeff Daily's avatar
Jeff Daily committed
1741
          requires:
1742
1743
          - nightly_binary_linux_wheel_py3.6_rocm4.1
          subfolder: rocm4.1/
Jeff Daily's avatar
Jeff Daily committed
1744
1745
1746
1747
1748
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1749
          name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
1750
1751
          python_version: '3.6'
          requires:
1752
          - nightly_binary_linux_wheel_py3.6_rocm4.1_upload
1753
      - binary_linux_wheel:
1754
          cu_version: rocm4.2
1755
1756
1757
1758
1759
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1760
          name: nightly_binary_linux_wheel_py3.6_rocm4.2
1761
          python_version: '3.6'
1762
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1763
1764
1765
1766
1767
1768
1769
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1770
          name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
1771
          requires:
1772
1773
          - nightly_binary_linux_wheel_py3.6_rocm4.2
          subfolder: rocm4.2/
1774
1775
1776
1777
1778
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1779
          name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
1780
1781
          python_version: '3.6'
          requires:
1782
          - nightly_binary_linux_wheel_py3.6_rocm4.2_upload
1783
      - binary_linux_wheel:
1784
          conda_docker_image: pytorch/conda-builder:cpu
1785
          cu_version: cpu
1786
1787
1788
          filters:
            branches:
              only: nightly
1789
1790
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
1791
          name: nightly_binary_linux_wheel_py3.7_cpu
1792
          python_version: '3.7'
1793
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
1794
1795
      - binary_wheel_upload:
          context: org-member
1796
1797
1798
          filters:
            branches:
              only: nightly
1799
1800
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1801
          name: nightly_binary_linux_wheel_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
1802
          requires:
1803
1804
          - nightly_binary_linux_wheel_py3.7_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
1805
1806
1807
1808
1809
1810
1811
1812
1813
      - 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
1814
      - binary_linux_wheel:
1815
          conda_docker_image: pytorch/conda-builder:cuda102
1816
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
1817
1818
1819
          filters:
            branches:
              only: nightly
1820
1821
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1822
          name: nightly_binary_linux_wheel_py3.7_cu102
Francisco Massa's avatar
Francisco Massa committed
1823
          python_version: '3.7'
1824
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
1825
1826
1827
1828
1829
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1830
1831
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1832
          name: nightly_binary_linux_wheel_py3.7_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
1833
          requires:
1834
1835
          - nightly_binary_linux_wheel_py3.7_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
1836
1837
1838
1839
1840
1841
1842
1843
1844
      - 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
1845
      - binary_linux_wheel:
1846
1847
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 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.7_cu111
peterjc123's avatar
peterjc123 committed
1854
          python_version: '3.7'
1855
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 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.7_cu111_upload
peterjc123's avatar
peterjc123 committed
1864
          requires:
1865
1866
          - nightly_binary_linux_wheel_py3.7_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
1867
1868
1869
1870
1871
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1872
          name: nightly_binary_linux_wheel_py3.7_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
1873
1874
          python_version: '3.7'
          requires:
1875
          - nightly_binary_linux_wheel_py3.7_cu111_upload
Jeff Daily's avatar
Jeff Daily committed
1876
      - binary_linux_wheel:
1877
          cu_version: rocm4.1
Jeff Daily's avatar
Jeff Daily committed
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.7_rocm4.1
Jeff Daily's avatar
Jeff Daily committed
1884
          python_version: '3.7'
1885
          wheel_docker_image: pytorch/manylinux-rocm:4.1
Jeff Daily's avatar
Jeff Daily committed
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.7_rocm4.1_upload
Jeff Daily's avatar
Jeff Daily committed
1894
          requires:
1895
1896
          - nightly_binary_linux_wheel_py3.7_rocm4.1
          subfolder: rocm4.1/
Jeff Daily's avatar
Jeff Daily committed
1897
1898
1899
1900
1901
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1902
          name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
1903
1904
          python_version: '3.7'
          requires:
1905
          - nightly_binary_linux_wheel_py3.7_rocm4.1_upload
1906
      - binary_linux_wheel:
1907
          cu_version: rocm4.2
1908
1909
1910
1911
1912
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1913
          name: nightly_binary_linux_wheel_py3.7_rocm4.2
1914
          python_version: '3.7'
1915
          wheel_docker_image: pytorch/manylinux-rocm:4.2
1916
1917
1918
1919
1920
1921
1922
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1923
          name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
1924
          requires:
1925
1926
          - nightly_binary_linux_wheel_py3.7_rocm4.2
          subfolder: rocm4.2/
1927
1928
1929
1930
1931
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
1932
          name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
1933
1934
          python_version: '3.7'
          requires:
1935
          - nightly_binary_linux_wheel_py3.7_rocm4.2_upload
1936
      - binary_linux_wheel:
1937
          conda_docker_image: pytorch/conda-builder:cpu
1938
1939
1940
1941
          cu_version: cpu
          filters:
            branches:
              only: nightly
1942
1943
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1944
1945
          name: nightly_binary_linux_wheel_py3.8_cpu
          python_version: '3.8'
1946
          wheel_docker_image: pytorch/manylinux-cuda102
1947
1948
1949
1950
1951
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1952
1953
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1954
1955
1956
1957
          name: nightly_binary_linux_wheel_py3.8_cpu_upload
          requires:
          - nightly_binary_linux_wheel_py3.8_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
1958
1959
1960
1961
1962
1963
1964
1965
1966
      - 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
1967
      - binary_linux_wheel:
1968
          conda_docker_image: pytorch/conda-builder:cuda102
1969
          cu_version: cu102
1970
1971
1972
          filters:
            branches:
              only: nightly
1973
1974
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1975
          name: nightly_binary_linux_wheel_py3.8_cu102
1976
          python_version: '3.8'
1977
          wheel_docker_image: pytorch/manylinux-cuda102
1978
1979
1980
1981
1982
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
1983
1984
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
1985
          name: nightly_binary_linux_wheel_py3.8_cu102_upload
1986
          requires:
1987
1988
          - nightly_binary_linux_wheel_py3.8_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
1989
1990
1991
1992
1993
1994
1995
1996
1997
      - 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
1998
      - binary_linux_wheel:
1999
2000
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2001
2002
2003
2004
2005
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2006
          name: nightly_binary_linux_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
2007
          python_version: '3.8'
2008
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
2009
2010
2011
2012
2013
2014
2015
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2016
          name: nightly_binary_linux_wheel_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
2017
          requires:
2018
2019
          - nightly_binary_linux_wheel_py3.8_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2020
2021
2022
2023
2024
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2025
          name: nightly_binary_linux_wheel_py3.8_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2026
2027
          python_version: '3.8'
          requires:
2028
          - nightly_binary_linux_wheel_py3.8_cu111_upload
Jeff Daily's avatar
Jeff Daily committed
2029
      - binary_linux_wheel:
2030
          cu_version: rocm4.1
Jeff Daily's avatar
Jeff Daily committed
2031
2032
2033
2034
2035
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2036
          name: nightly_binary_linux_wheel_py3.8_rocm4.1
Jeff Daily's avatar
Jeff Daily committed
2037
          python_version: '3.8'
2038
          wheel_docker_image: pytorch/manylinux-rocm:4.1
Jeff Daily's avatar
Jeff Daily committed
2039
2040
2041
2042
2043
2044
2045
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2046
          name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload
Jeff Daily's avatar
Jeff Daily committed
2047
          requires:
2048
2049
          - nightly_binary_linux_wheel_py3.8_rocm4.1
          subfolder: rocm4.1/
Jeff Daily's avatar
Jeff Daily committed
2050
2051
2052
2053
2054
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2055
          name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
2056
2057
          python_version: '3.8'
          requires:
2058
          - nightly_binary_linux_wheel_py3.8_rocm4.1_upload
2059
      - binary_linux_wheel:
2060
          cu_version: rocm4.2
2061
2062
2063
2064
2065
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2066
          name: nightly_binary_linux_wheel_py3.8_rocm4.2
2067
          python_version: '3.8'
2068
          wheel_docker_image: pytorch/manylinux-rocm:4.2
2069
2070
2071
2072
2073
2074
2075
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2076
          name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
2077
          requires:
2078
2079
          - nightly_binary_linux_wheel_py3.8_rocm4.2
          subfolder: rocm4.2/
2080
2081
2082
2083
2084
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2085
          name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
2086
2087
          python_version: '3.8'
          requires:
2088
          - nightly_binary_linux_wheel_py3.8_rocm4.2_upload
2089
2090
      - binary_linux_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
2091
          cu_version: cpu
2092
2093
2094
          filters:
            branches:
              only: nightly
2095
2096
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2097
2098
          name: nightly_binary_linux_wheel_py3.9_cpu
          python_version: '3.9'
2099
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
2100
2101
      - binary_wheel_upload:
          context: org-member
2102
2103
2104
          filters:
            branches:
              only: nightly
2105
2106
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2107
          name: nightly_binary_linux_wheel_py3.9_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2108
          requires:
2109
2110
2111
2112
2113
2114
2115
2116
2117
2118
2119
2120
2121
2122
          - 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
2123
2124
2125
          filters:
            branches:
              only: nightly
2126
2127
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2128
2129
          name: nightly_binary_linux_wheel_py3.9_cu102
          python_version: '3.9'
2130
          wheel_docker_image: pytorch/manylinux-cuda102
2131
2132
2133
2134
2135
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2136
2137
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2138
          name: nightly_binary_linux_wheel_py3.9_cu102_upload
2139
          requires:
2140
2141
2142
2143
2144
2145
2146
2147
2148
2149
2150
2151
          - 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:
2152
2153
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
2154
2155
2156
          filters:
            branches:
              only: nightly
2157
2158
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2159
          name: nightly_binary_linux_wheel_py3.9_cu111
2160
          python_version: '3.9'
2161
          wheel_docker_image: pytorch/manylinux-cuda111
2162
2163
2164
2165
2166
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2167
2168
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2169
          name: nightly_binary_linux_wheel_py3.9_cu111_upload
2170
          requires:
2171
2172
          - nightly_binary_linux_wheel_py3.9_cu111
          subfolder: cu111/
2173
      - smoke_test_linux_pip:
guyang3532's avatar
guyang3532 committed
2174
2175
2176
2177
          filters:
            branches:
              only:
              - nightly
2178
          name: nightly_binary_linux_wheel_py3.9_cu111_smoke_test_pip
2179
          python_version: '3.9'
guyang3532's avatar
guyang3532 committed
2180
          requires:
2181
          - nightly_binary_linux_wheel_py3.9_cu111_upload
Jeff Daily's avatar
Jeff Daily committed
2182
      - binary_linux_wheel:
2183
          cu_version: rocm4.1
Jeff Daily's avatar
Jeff Daily committed
2184
2185
2186
2187
2188
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2189
          name: nightly_binary_linux_wheel_py3.9_rocm4.1
Jeff Daily's avatar
Jeff Daily committed
2190
          python_version: '3.9'
2191
          wheel_docker_image: pytorch/manylinux-rocm:4.1
Jeff Daily's avatar
Jeff Daily committed
2192
2193
2194
2195
2196
2197
2198
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2199
          name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload
Jeff Daily's avatar
Jeff Daily committed
2200
          requires:
2201
2202
          - nightly_binary_linux_wheel_py3.9_rocm4.1
          subfolder: rocm4.1/
Jeff Daily's avatar
Jeff Daily committed
2203
2204
2205
2206
2207
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2208
          name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip
Jeff Daily's avatar
Jeff Daily committed
2209
2210
          python_version: '3.9'
          requires:
2211
          - nightly_binary_linux_wheel_py3.9_rocm4.1_upload
2212
      - binary_linux_wheel:
2213
          cu_version: rocm4.2
2214
2215
2216
2217
2218
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2219
          name: nightly_binary_linux_wheel_py3.9_rocm4.2
2220
          python_version: '3.9'
2221
          wheel_docker_image: pytorch/manylinux-rocm:4.2
2222
2223
2224
2225
2226
2227
2228
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2229
          name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
2230
          requires:
2231
2232
          - nightly_binary_linux_wheel_py3.9_rocm4.2
          subfolder: rocm4.2/
2233
2234
2235
2236
2237
      - smoke_test_linux_pip:
          filters:
            branches:
              only:
              - nightly
2238
          name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
2239
2240
          python_version: '3.9'
          requires:
2241
          - nightly_binary_linux_wheel_py3.9_rocm4.2_upload
2242
2243
2244
      - binary_macos_wheel:
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
2245
2246
2247
          filters:
            branches:
              only: nightly
2248
2249
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2250
          name: nightly_binary_macos_wheel_py3.6_cpu
2251
          python_version: '3.6'
2252
          wheel_docker_image: pytorch/manylinux-cuda102
2253
2254
2255
2256
2257
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2258
2259
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2260
          name: nightly_binary_macos_wheel_py3.6_cpu_upload
2261
          requires:
2262
2263
2264
2265
2266
2267
2268
2269
2270
2271
2272
2273
2274
2275
2276
2277
2278
2279
2280
2281
2282
2283
2284
2285
2286
2287
2288
2289
2290
2291
2292
2293
2294
2295
2296
2297
2298
2299
2300
2301
2302
2303
2304
2305
2306
2307
2308
2309
2310
2311
2312
2313
2314
2315
2316
2317
2318
2319
2320
2321
2322
2323
2324
2325
2326
2327
2328
2329
2330
2331
2332
2333
2334
2335
2336
2337
2338
2339
2340
2341
2342
2343
2344
2345
2346
2347
2348
2349
2350
2351
2352
2353
2354
2355
2356
2357
2358
          - 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
2359
      - binary_win_wheel:
2360
          cu_version: cu102
2361
2362
2363
          filters:
            branches:
              only: nightly
2364
2365
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2366
2367
2368
2369
2370
2371
2372
          name: nightly_binary_win_wheel_py3.6_cu102
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2373
2374
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2375
2376
2377
2378
          name: nightly_binary_win_wheel_py3.6_cu102_upload
          requires:
          - nightly_binary_win_wheel_py3.6_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
2379
2380
2381
2382
2383
2384
2385
2386
2387
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.6_cu102_smoke_test_pip
          python_version: '3.6'
          requires:
          - nightly_binary_win_wheel_py3.6_cu102_upload
peterjc123's avatar
peterjc123 committed
2388
      - binary_win_wheel:
2389
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2390
2391
2392
2393
2394
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2395
          name: nightly_binary_win_wheel_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2396
2397
2398
2399
2400
2401
2402
2403
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2404
          name: nightly_binary_win_wheel_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
2405
          requires:
2406
2407
          - nightly_binary_win_wheel_py3.6_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2408
2409
2410
2411
2412
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2413
          name: nightly_binary_win_wheel_py3.6_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2414
2415
          python_version: '3.6'
          requires:
2416
          - nightly_binary_win_wheel_py3.6_cu111_upload
2417
      - binary_win_wheel:
2418
2419
2420
2421
          cu_version: cpu
          filters:
            branches:
              only: nightly
2422
2423
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2424
2425
2426
2427
2428
2429
2430
          name: nightly_binary_win_wheel_py3.7_cpu
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2431
2432
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2433
2434
2435
2436
          name: nightly_binary_win_wheel_py3.7_cpu_upload
          requires:
          - nightly_binary_win_wheel_py3.7_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
2437
2438
2439
2440
2441
2442
2443
2444
2445
      - 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
2446
      - binary_win_wheel:
2447
          cu_version: cu102
2448
2449
2450
          filters:
            branches:
              only: nightly
2451
2452
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2453
2454
2455
2456
2457
2458
2459
          name: nightly_binary_win_wheel_py3.7_cu102
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2460
2461
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2462
2463
2464
2465
          name: nightly_binary_win_wheel_py3.7_cu102_upload
          requires:
          - nightly_binary_win_wheel_py3.7_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
2466
2467
2468
2469
2470
2471
2472
2473
2474
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.7_cu102_smoke_test_pip
          python_version: '3.7'
          requires:
          - nightly_binary_win_wheel_py3.7_cu102_upload
peterjc123's avatar
peterjc123 committed
2475
      - binary_win_wheel:
2476
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2477
2478
2479
2480
2481
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2482
          name: nightly_binary_win_wheel_py3.7_cu111
peterjc123's avatar
peterjc123 committed
2483
2484
2485
2486
2487
2488
2489
2490
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2491
          name: nightly_binary_win_wheel_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
2492
          requires:
2493
2494
          - nightly_binary_win_wheel_py3.7_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2495
2496
2497
2498
2499
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2500
          name: nightly_binary_win_wheel_py3.7_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2501
2502
          python_version: '3.7'
          requires:
2503
          - nightly_binary_win_wheel_py3.7_cu111_upload
2504
      - binary_win_wheel:
2505
2506
2507
2508
          cu_version: cpu
          filters:
            branches:
              only: nightly
2509
2510
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2511
2512
2513
2514
2515
2516
2517
          name: nightly_binary_win_wheel_py3.8_cpu
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2518
2519
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2520
2521
2522
2523
          name: nightly_binary_win_wheel_py3.8_cpu_upload
          requires:
          - nightly_binary_win_wheel_py3.8_cpu
          subfolder: cpu/
guyang3532's avatar
guyang3532 committed
2524
2525
2526
2527
2528
2529
2530
2531
2532
      - 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
2533
      - binary_win_wheel:
2534
          cu_version: cu102
2535
2536
2537
          filters:
            branches:
              only: nightly
2538
2539
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2540
2541
2542
2543
2544
2545
2546
          name: nightly_binary_win_wheel_py3.8_cu102
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2547
2548
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2549
2550
2551
2552
          name: nightly_binary_win_wheel_py3.8_cu102_upload
          requires:
          - nightly_binary_win_wheel_py3.8_cu102
          subfolder: cu102/
guyang3532's avatar
guyang3532 committed
2553
2554
2555
2556
2557
2558
2559
2560
2561
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.8_cu102_smoke_test_pip
          python_version: '3.8'
          requires:
          - nightly_binary_win_wheel_py3.8_cu102_upload
peterjc123's avatar
peterjc123 committed
2562
      - binary_win_wheel:
2563
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2564
2565
2566
2567
2568
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2569
          name: nightly_binary_win_wheel_py3.8_cu111
peterjc123's avatar
peterjc123 committed
2570
2571
2572
2573
2574
2575
2576
2577
          python_version: '3.8'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2578
          name: nightly_binary_win_wheel_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
2579
          requires:
2580
2581
          - nightly_binary_win_wheel_py3.8_cu111
          subfolder: cu111/
peterjc123's avatar
peterjc123 committed
2582
2583
2584
2585
2586
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2587
          name: nightly_binary_win_wheel_py3.8_cu111_smoke_test_pip
peterjc123's avatar
peterjc123 committed
2588
2589
          python_version: '3.8'
          requires:
2590
          - nightly_binary_win_wheel_py3.8_cu111_upload
2591
      - binary_win_wheel:
2592
          cu_version: cpu
2593
2594
2595
          filters:
            branches:
              only: nightly
2596
2597
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2598
2599
2600
          name: nightly_binary_win_wheel_py3.9_cpu
          python_version: '3.9'
      - binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
2601
          context: org-member
2602
2603
2604
          filters:
            branches:
              only: nightly
2605
2606
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2607
          name: nightly_binary_win_wheel_py3.9_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2608
          requires:
2609
2610
2611
          - nightly_binary_win_wheel_py3.9_cpu
          subfolder: cpu/
      - smoke_test_win_pip:
guyang3532's avatar
guyang3532 committed
2612
2613
2614
2615
          filters:
            branches:
              only:
              - nightly
2616
2617
          name: nightly_binary_win_wheel_py3.9_cpu_smoke_test_pip
          python_version: '3.9'
guyang3532's avatar
guyang3532 committed
2618
          requires:
2619
2620
2621
2622
2623
2624
2625
2626
2627
2628
2629
2630
2631
2632
2633
2634
2635
2636
2637
2638
2639
2640
2641
2642
2643
2644
2645
2646
2647
2648
2649
          - nightly_binary_win_wheel_py3.9_cpu_upload
      - binary_win_wheel:
          cu_version: cu102
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_wheel_py3.9_cu102
          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_cu102_upload
          requires:
          - nightly_binary_win_wheel_py3.9_cu102
          subfolder: cu102/
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_wheel_py3.9_cu102_smoke_test_pip
          python_version: '3.9'
          requires:
          - nightly_binary_win_wheel_py3.9_cu102_upload
      - binary_win_wheel:
2650
          cu_version: cu111
2651
2652
2653
2654
2655
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2656
          name: nightly_binary_win_wheel_py3.9_cu111
2657
2658
2659
2660
2661
2662
2663
2664
          python_version: '3.9'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2665
          name: nightly_binary_win_wheel_py3.9_cu111_upload
2666
          requires:
2667
2668
          - nightly_binary_win_wheel_py3.9_cu111
          subfolder: cu111/
2669
2670
2671
2672
2673
      - smoke_test_win_pip:
          filters:
            branches:
              only:
              - nightly
2674
          name: nightly_binary_win_wheel_py3.9_cu111_smoke_test_pip
2675
2676
          python_version: '3.9'
          requires:
2677
          - nightly_binary_win_wheel_py3.9_cu111_upload
2678
      - binary_linux_conda:
2679
2680
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
2681
2682
2683
          filters:
            branches:
              only: nightly
2684
2685
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2686
          name: nightly_binary_linux_conda_py3.6_cpu
2687
          python_version: '3.6'
2688
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
2689
2690
      - binary_conda_upload:
          context: org-member
2691
2692
2693
          filters:
            branches:
              only: nightly
2694
2695
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2696
          name: nightly_binary_linux_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2697
          requires:
2698
          - nightly_binary_linux_conda_py3.6_cpu
guyang3532's avatar
guyang3532 committed
2699
2700
2701
2702
2703
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2704
          name: nightly_binary_linux_conda_py3.6_cpu_smoke_test_conda
guyang3532's avatar
guyang3532 committed
2705
2706
          python_version: '3.6'
          requires:
2707
          - nightly_binary_linux_conda_py3.6_cpu_upload
Francisco Massa's avatar
Francisco Massa committed
2708
      - binary_linux_conda:
2709
          conda_docker_image: pytorch/conda-builder:cuda102
2710
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
2711
2712
2713
          filters:
            branches:
              only: nightly
2714
2715
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2716
          name: nightly_binary_linux_conda_py3.6_cu102
Francisco Massa's avatar
Francisco Massa committed
2717
          python_version: '3.6'
2718
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
2719
2720
2721
2722
2723
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2724
2725
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2726
          name: nightly_binary_linux_conda_py3.6_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
2727
          requires:
2728
          - nightly_binary_linux_conda_py3.6_cu102
guyang3532's avatar
guyang3532 committed
2729
2730
2731
2732
2733
2734
2735
2736
2737
      - 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
2738
      - binary_linux_conda:
2739
2740
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2741
2742
2743
2744
2745
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2746
          name: nightly_binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2747
          python_version: '3.6'
2748
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
2749
2750
2751
2752
2753
2754
2755
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2756
          name: nightly_binary_linux_conda_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
2757
          requires:
2758
          - nightly_binary_linux_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
2759
2760
2761
2762
2763
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2764
          name: nightly_binary_linux_conda_py3.6_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
2765
2766
          python_version: '3.6'
          requires:
2767
          - nightly_binary_linux_conda_py3.6_cu111_upload
2768
      - binary_linux_conda:
2769
          conda_docker_image: pytorch/conda-builder:cpu
2770
          cu_version: cpu
2771
2772
2773
          filters:
            branches:
              only: nightly
2774
2775
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
2776
          name: nightly_binary_linux_conda_py3.7_cpu
2777
          python_version: '3.7'
2778
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
2779
2780
      - binary_conda_upload:
          context: org-member
2781
2782
2783
          filters:
            branches:
              only: nightly
2784
2785
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2786
          name: nightly_binary_linux_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
2787
          requires:
2788
          - nightly_binary_linux_conda_py3.7_cpu
guyang3532's avatar
guyang3532 committed
2789
2790
2791
2792
2793
2794
2795
2796
2797
      - 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
2798
      - binary_linux_conda:
2799
          conda_docker_image: pytorch/conda-builder:cuda102
2800
          cu_version: cu102
Francisco Massa's avatar
Francisco Massa committed
2801
2802
2803
          filters:
            branches:
              only: nightly
2804
2805
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2806
          name: nightly_binary_linux_conda_py3.7_cu102
Francisco Massa's avatar
Francisco Massa committed
2807
          python_version: '3.7'
2808
          wheel_docker_image: pytorch/manylinux-cuda102
Francisco Massa's avatar
Francisco Massa committed
2809
2810
2811
2812
2813
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2814
2815
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2816
          name: nightly_binary_linux_conda_py3.7_cu102_upload
Francisco Massa's avatar
Francisco Massa committed
2817
          requires:
2818
          - nightly_binary_linux_conda_py3.7_cu102
guyang3532's avatar
guyang3532 committed
2819
2820
2821
2822
2823
2824
2825
2826
2827
      - 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
2828
      - binary_linux_conda:
2829
2830
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
2831
2832
2833
2834
2835
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2836
          name: nightly_binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
2837
          python_version: '3.7'
2838
          wheel_docker_image: pytorch/manylinux-cuda111
peterjc123's avatar
peterjc123 committed
2839
2840
2841
2842
2843
2844
2845
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2846
          name: nightly_binary_linux_conda_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
2847
          requires:
2848
          - nightly_binary_linux_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
2849
2850
2851
2852
2853
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2854
          name: nightly_binary_linux_conda_py3.7_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
2855
2856
          python_version: '3.7'
          requires:
2857
          - nightly_binary_linux_conda_py3.7_cu111_upload
2858
      - binary_linux_conda:
2859
          conda_docker_image: pytorch/conda-builder:cpu
2860
2861
2862
2863
          cu_version: cpu
          filters:
            branches:
              only: nightly
2864
2865
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2866
2867
          name: nightly_binary_linux_conda_py3.8_cpu
          python_version: '3.8'
2868
          wheel_docker_image: pytorch/manylinux-cuda102
2869
2870
2871
2872
2873
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2874
2875
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2876
2877
2878
          name: nightly_binary_linux_conda_py3.8_cpu_upload
          requires:
          - nightly_binary_linux_conda_py3.8_cpu
guyang3532's avatar
guyang3532 committed
2879
2880
2881
2882
2883
2884
2885
2886
2887
      - 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
2888
      - binary_linux_conda:
2889
2890
          conda_docker_image: pytorch/conda-builder:cuda102
          cu_version: cu102
2891
2892
2893
          filters:
            branches:
              only: nightly
2894
2895
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2896
          name: nightly_binary_linux_conda_py3.8_cu102
2897
          python_version: '3.8'
2898
          wheel_docker_image: pytorch/manylinux-cuda102
2899
2900
2901
2902
2903
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2904
2905
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2906
          name: nightly_binary_linux_conda_py3.8_cu102_upload
2907
          requires:
2908
          - nightly_binary_linux_conda_py3.8_cu102
guyang3532's avatar
guyang3532 committed
2909
2910
2911
2912
2913
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2914
          name: nightly_binary_linux_conda_py3.8_cu102_smoke_test_conda
guyang3532's avatar
guyang3532 committed
2915
2916
          python_version: '3.8'
          requires:
2917
          - nightly_binary_linux_conda_py3.8_cu102_upload
2918
      - binary_linux_conda:
2919
2920
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
2921
2922
2923
          filters:
            branches:
              only: nightly
2924
2925
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2926
          name: nightly_binary_linux_conda_py3.8_cu111
2927
          python_version: '3.8'
2928
          wheel_docker_image: pytorch/manylinux-cuda111
2929
2930
2931
2932
2933
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
2934
2935
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2936
          name: nightly_binary_linux_conda_py3.8_cu111_upload
2937
          requires:
2938
          - nightly_binary_linux_conda_py3.8_cu111
guyang3532's avatar
guyang3532 committed
2939
2940
2941
2942
2943
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2944
          name: nightly_binary_linux_conda_py3.8_cu111_smoke_test_conda
guyang3532's avatar
guyang3532 committed
2945
2946
          python_version: '3.8'
          requires:
2947
          - nightly_binary_linux_conda_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
2948
      - binary_linux_conda:
2949
2950
          conda_docker_image: pytorch/conda-builder:cpu
          cu_version: cpu
peterjc123's avatar
peterjc123 committed
2951
2952
2953
2954
2955
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2956
2957
2958
          name: nightly_binary_linux_conda_py3.9_cpu
          python_version: '3.9'
          wheel_docker_image: pytorch/manylinux-cuda102
peterjc123's avatar
peterjc123 committed
2959
2960
2961
2962
2963
2964
2965
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
2966
          name: nightly_binary_linux_conda_py3.9_cpu_upload
peterjc123's avatar
peterjc123 committed
2967
          requires:
2968
          - nightly_binary_linux_conda_py3.9_cpu
peterjc123's avatar
peterjc123 committed
2969
2970
2971
2972
2973
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
2974
2975
2976
2977
2978
2979
2980
2981
2982
2983
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
          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:
3009
3010
          conda_docker_image: pytorch/conda-builder:cuda111
          cu_version: cu111
3011
3012
3013
3014
3015
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3016
          name: nightly_binary_linux_conda_py3.9_cu111
3017
          python_version: '3.9'
3018
          wheel_docker_image: pytorch/manylinux-cuda111
3019
3020
3021
3022
3023
3024
3025
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3026
          name: nightly_binary_linux_conda_py3.9_cu111_upload
peterjc123's avatar
peterjc123 committed
3027
          requires:
3028
          - nightly_binary_linux_conda_py3.9_cu111
3029
3030
3031
3032
3033
      - smoke_test_linux_conda:
          filters:
            branches:
              only:
              - nightly
3034
          name: nightly_binary_linux_conda_py3.9_cu111_smoke_test_conda
3035
3036
          python_version: '3.9'
          requires:
3037
          - nightly_binary_linux_conda_py3.9_cu111_upload
3038
      - binary_macos_conda:
3039
          conda_docker_image: pytorch/conda-builder:cpu
3040
          cu_version: cpu
3041
3042
3043
          filters:
            branches:
              only: nightly
3044
3045
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
3046
          name: nightly_binary_macos_conda_py3.6_cpu
3047
          python_version: '3.6'
3048
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
3049
3050
      - binary_conda_upload:
          context: org-member
3051
3052
3053
          filters:
            branches:
              only: nightly
3054
3055
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3056
          name: nightly_binary_macos_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
3057
          requires:
3058
          - nightly_binary_macos_conda_py3.6_cpu
3059
      - binary_macos_conda:
3060
          conda_docker_image: pytorch/conda-builder:cpu
3061
          cu_version: cpu
3062
3063
3064
          filters:
            branches:
              only: nightly
3065
3066
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
Edward Z. Yang's avatar
Edward Z. Yang committed
3067
          name: nightly_binary_macos_conda_py3.7_cpu
3068
          python_version: '3.7'
3069
          wheel_docker_image: pytorch/manylinux-cuda102
Edward Z. Yang's avatar
Edward Z. Yang committed
3070
3071
      - binary_conda_upload:
          context: org-member
3072
3073
3074
          filters:
            branches:
              only: nightly
3075
3076
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3077
          name: nightly_binary_macos_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
3078
          requires:
3079
3080
          - nightly_binary_macos_conda_py3.7_cpu
      - binary_macos_conda:
3081
          conda_docker_image: pytorch/conda-builder:cpu
3082
3083
3084
3085
          cu_version: cpu
          filters:
            branches:
              only: nightly
3086
3087
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3088
3089
          name: nightly_binary_macos_conda_py3.8_cpu
          python_version: '3.8'
3090
          wheel_docker_image: pytorch/manylinux-cuda102
3091
3092
3093
3094
3095
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3096
3097
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3098
3099
          name: nightly_binary_macos_conda_py3.8_cpu_upload
          requires:
3100
          - nightly_binary_macos_conda_py3.8_cpu
3101
3102
3103
3104
3105
3106
3107
3108
3109
3110
3111
3112
3113
3114
3115
3116
3117
3118
3119
3120
3121
      - 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
3122
      - binary_win_conda:
3123
3124
3125
3126
          cu_version: cpu
          filters:
            branches:
              only: nightly
3127
3128
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3129
3130
3131
3132
3133
3134
3135
          name: nightly_binary_win_conda_py3.6_cpu
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3136
3137
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3138
3139
3140
          name: nightly_binary_win_conda_py3.6_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.6_cpu
guyang3532's avatar
guyang3532 committed
3141
3142
3143
3144
3145
3146
3147
3148
3149
      - 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
3150
      - binary_win_conda:
3151
          cu_version: cu102
3152
3153
3154
          filters:
            branches:
              only: nightly
3155
3156
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3157
3158
3159
3160
3161
3162
3163
          name: nightly_binary_win_conda_py3.6_cu102
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3164
3165
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3166
3167
3168
          name: nightly_binary_win_conda_py3.6_cu102_upload
          requires:
          - nightly_binary_win_conda_py3.6_cu102
guyang3532's avatar
guyang3532 committed
3169
3170
3171
3172
3173
3174
3175
3176
3177
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.6_cu102_smoke_test_conda
          python_version: '3.6'
          requires:
          - nightly_binary_win_conda_py3.6_cu102_upload
peterjc123's avatar
peterjc123 committed
3178
      - binary_win_conda:
3179
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3180
3181
3182
3183
3184
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3185
          name: nightly_binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
3186
3187
3188
3189
3190
3191
3192
3193
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3194
          name: nightly_binary_win_conda_py3.6_cu111_upload
peterjc123's avatar
peterjc123 committed
3195
          requires:
3196
          - nightly_binary_win_conda_py3.6_cu111
peterjc123's avatar
peterjc123 committed
3197
3198
3199
3200
3201
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3202
          name: nightly_binary_win_conda_py3.6_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3203
3204
          python_version: '3.6'
          requires:
3205
          - nightly_binary_win_conda_py3.6_cu111_upload
3206
      - binary_win_conda:
3207
3208
3209
3210
          cu_version: cpu
          filters:
            branches:
              only: nightly
3211
3212
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3213
3214
3215
3216
3217
3218
3219
          name: nightly_binary_win_conda_py3.7_cpu
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3220
3221
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3222
3223
3224
          name: nightly_binary_win_conda_py3.7_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.7_cpu
guyang3532's avatar
guyang3532 committed
3225
3226
3227
3228
3229
3230
3231
3232
3233
      - 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
3234
      - binary_win_conda:
3235
          cu_version: cu102
3236
3237
3238
          filters:
            branches:
              only: nightly
3239
3240
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3241
3242
3243
3244
3245
3246
3247
          name: nightly_binary_win_conda_py3.7_cu102
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3248
3249
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3250
3251
3252
          name: nightly_binary_win_conda_py3.7_cu102_upload
          requires:
          - nightly_binary_win_conda_py3.7_cu102
guyang3532's avatar
guyang3532 committed
3253
3254
3255
3256
3257
3258
3259
3260
3261
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.7_cu102_smoke_test_conda
          python_version: '3.7'
          requires:
          - nightly_binary_win_conda_py3.7_cu102_upload
peterjc123's avatar
peterjc123 committed
3262
      - binary_win_conda:
3263
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3264
3265
3266
3267
3268
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3269
          name: nightly_binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3270
3271
3272
3273
3274
3275
3276
3277
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3278
          name: nightly_binary_win_conda_py3.7_cu111_upload
peterjc123's avatar
peterjc123 committed
3279
          requires:
3280
          - nightly_binary_win_conda_py3.7_cu111
peterjc123's avatar
peterjc123 committed
3281
3282
3283
3284
3285
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3286
          name: nightly_binary_win_conda_py3.7_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3287
3288
          python_version: '3.7'
          requires:
3289
          - nightly_binary_win_conda_py3.7_cu111_upload
3290
      - binary_win_conda:
3291
3292
3293
3294
          cu_version: cpu
          filters:
            branches:
              only: nightly
3295
3296
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3297
3298
3299
3300
3301
3302
3303
          name: nightly_binary_win_conda_py3.8_cpu
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3304
3305
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3306
3307
3308
          name: nightly_binary_win_conda_py3.8_cpu_upload
          requires:
          - nightly_binary_win_conda_py3.8_cpu
guyang3532's avatar
guyang3532 committed
3309
3310
3311
3312
3313
3314
3315
3316
3317
      - 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
3318
      - binary_win_conda:
3319
          cu_version: cu102
3320
3321
3322
          filters:
            branches:
              only: nightly
3323
3324
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3325
3326
3327
3328
3329
3330
3331
          name: nightly_binary_win_conda_py3.8_cu102
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
3332
3333
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3334
3335
          name: nightly_binary_win_conda_py3.8_cu102_upload
          requires:
guyang3532's avatar
guyang3532 committed
3336
3337
3338
3339
3340
3341
3342
3343
3344
3345
          - nightly_binary_win_conda_py3.8_cu102
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.8_cu102_smoke_test_conda
          python_version: '3.8'
          requires:
          - nightly_binary_win_conda_py3.8_cu102_upload
peterjc123's avatar
peterjc123 committed
3346
      - binary_win_conda:
3347
          cu_version: cu111
peterjc123's avatar
peterjc123 committed
3348
3349
3350
3351
3352
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3353
          name: nightly_binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
3354
3355
3356
3357
3358
3359
3360
3361
          python_version: '3.8'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3362
          name: nightly_binary_win_conda_py3.8_cu111_upload
peterjc123's avatar
peterjc123 committed
3363
          requires:
3364
          - nightly_binary_win_conda_py3.8_cu111
peterjc123's avatar
peterjc123 committed
3365
3366
3367
3368
3369
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3370
          name: nightly_binary_win_conda_py3.8_cu111_smoke_test_conda
peterjc123's avatar
peterjc123 committed
3371
3372
          python_version: '3.8'
          requires:
3373
          - nightly_binary_win_conda_py3.8_cu111_upload
3374
3375
3376
3377
3378
3379
3380
3381
3382
3383
3384
3385
3386
3387
3388
3389
3390
3391
3392
3393
3394
3395
3396
3397
3398
3399
3400
3401
3402
3403
3404
3405
3406
3407
3408
3409
3410
3411
3412
3413
3414
3415
3416
3417
3418
3419
3420
3421
3422
3423
3424
3425
3426
3427
3428
3429
3430
      - 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:
          cu_version: cu102
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
          name: nightly_binary_win_conda_py3.9_cu102
          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_cu102_upload
          requires:
          - nightly_binary_win_conda_py3.9_cu102
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
          name: nightly_binary_win_conda_py3.9_cu102_smoke_test_conda
          python_version: '3.9'
          requires:
          - nightly_binary_win_conda_py3.9_cu102_upload
      - binary_win_conda:
3431
          cu_version: cu111
3432
3433
3434
3435
3436
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3437
          name: nightly_binary_win_conda_py3.9_cu111
3438
3439
3440
3441
3442
3443
3444
3445
          python_version: '3.9'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
            tags:
              only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
3446
          name: nightly_binary_win_conda_py3.9_cu111_upload
3447
          requires:
3448
          - nightly_binary_win_conda_py3.9_cu111
3449
3450
3451
3452
3453
      - smoke_test_win_conda:
          filters:
            branches:
              only:
              - nightly
3454
          name: nightly_binary_win_conda_py3.9_cu111_smoke_test_conda
3455
3456
          python_version: '3.9'
          requires:
3457
          - nightly_binary_win_conda_py3.9_cu111_upload
guyang3532's avatar
guyang3532 committed
3458
3459
3460
3461
3462
3463
3464
3465
3466
3467
  docker_build:
    triggers:
      - schedule:
          cron: "0 10 * * 0"
          filters:
            branches:
              only:
                - master
    jobs:
      - smoke_test_docker_image_build:
3468
          context: org-member