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

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

  cmake:
    jobs:
      - cmake_linux_cpu:
          cu_version: cpu
          name: cmake_linux_cpu
          python_version: '3.8'
      - cmake_linux_gpu:
1579
          cu_version: cu102
1580
1581
          name: cmake_linux_gpu
          python_version: '3.8'
1582
          wheel_docker_image: pytorch/manylinux-cuda102
1583
1584
1585
1586
      - cmake_windows_cpu:
          cu_version: cpu
          name: cmake_windows_cpu
          python_version: '3.8'
1587
      - cmake_windows_gpu:
1588
          cu_version: cu102
1589
1590
          name: cmake_windows_gpu
          python_version: '3.8'
1591
1592
1593
1594
1595
      - cmake_macos_cpu:
          cu_version: cpu
          name: cmake_macos_cpu
          python_version: '3.8'

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