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

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

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

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

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

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

147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
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 >>

163
164
165
166
167
168
169
170
torchvision_android_params: &torchvision_android_params
  parameters:
    build_environment:
      type: string
      default: ""
  environment:
    BUILD_ENVIRONMENT: << parameters.build_environment >>

guyang3532's avatar
guyang3532 committed
171
172
173
174
175
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
    - image: torchvision/smoke_test:latest

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

188
189
190
191
192
193
194
  python_lint:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
195
196
197
            pip install --user --progress-bar off pre-commit
            pre-commit install-hooks
      - run: pre-commit run --all-files
198
199
200
201
202
203
204
205

  python_type_check:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
206
207
            sudo apt-get update -y
            sudo apt install -y libturbojpeg-dev
208
            pip install --user --progress-bar off mypy
209
            pip install --user --progress-bar off types-requests
210
            pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
211
            pip install --user --progress-bar off --editable .
212
            mypy --config-file mypy.ini
213

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

224
225
226
227
228
229
230
  clang_format:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
231
232
233
            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
234
            ./.circleci/unittest/linux/scripts/run-clang-format.py -r torchvision/csrc --clang-format-executable /opt/clang-format
235

Francisco Massa's avatar
Francisco Massa committed
236
237
238
239
240
241
242
243
244
  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
245
            pip install --user --progress-bar off .
Anirudh's avatar
Anirudh committed
246
            pip install pytest
Francisco Massa's avatar
Francisco Massa committed
247
248
            python test/test_hub.py

249
250
251
252
253
254
255
256
257
  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
258
            pip install --user --progress-bar off .
259
            pip install --user onnx
260
            pip install --user onnxruntime
261
            pip install --user pytest
262
263
            python test/test_onnx.py

264
265
266
  binary_linux_wheel:
    <<: *binary_common
    docker:
Edward Z. Yang's avatar
Edward Z. Yang committed
267
      - image: << parameters.wheel_docker_image >>
268
269
    resource_class: 2xlarge+
    steps:
270
      - checkout_merge
271
      - designate_upload_channel
272
273
274
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
275
276
277
278
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
279
280
281
282

  binary_linux_conda:
    <<: *binary_common
    docker:
283
      - image: "<< parameters.conda_docker_image >>"
284
    resource_class: 2xlarge+
285
    steps:
286
      - checkout_merge
287
      - designate_upload_channel
288
289
290
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
291
292
293
294
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"
295
296
      - store_test_results:
          path: build_results/
297

Francisco Massa's avatar
Francisco Massa committed
298
  binary_win_conda:
299
300
    <<: *binary_common
    executor: windows-cpu
301
302
    steps:
      - checkout_merge
303
      - designate_upload_channel
304
      - install_cuda_compatible_cmath
305
306
      - run:
          name: Build conda packages
307
          no_output_timeout: 20m
308
          command: |
309
310
311
312
313
314
            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"
315
316
317
318
            # cudatoolkit >= 11 isn't available for windows in the nvidia channel
            if [[ "${CU_VERSION}" =~ cu11.* ]]; then
              export CONDA_CHANNEL_FLAGS="-c conda-forge"
            fi
319
            packaging/build_conda.sh
320
            rm /C/tools/miniconda3/conda-bld/win-64/vs${VC_YEAR}*.tar.bz2
321
      - store_artifacts:
322
          path: C:/tools/miniconda3/conda-bld/win-64
323
      - persist_to_workspace:
324
          root: C:/tools/miniconda3/conda-bld/win-64
325
326
327
328
329
          paths:
            - "*"
      - store_test_results:
          path: build_results/

330
  binary_win_wheel:
331
332
    <<: *binary_common
    executor: windows-cpu
333
334
    steps:
      - checkout_merge
335
      - designate_upload_channel
336
      - install_cuda_compatible_cmath
337
338
339
      - run:
          name: Build wheel packages
          command: |
340
341
342
343
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_wheel.sh
344
      - store_artifacts:
345
          path: dist
346
      - persist_to_workspace:
347
          root: dist
348
349
350
351
352
          paths:
            - "*"
      - store_test_results:
          path: build_results/

353
354
355
  binary_macos_wheel:
    <<: *binary_common
    macos:
356
      xcode: "12.0"
357
    steps:
358
      - checkout_merge
359
      - designate_upload_channel
360
361
362
363
364
365
366
367
368
369
370
      - 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
371
372
373
374
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
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
401
402
403
404
405
406
407
408
409
410
411
412
  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"

413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
  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"

449
450
451
  binary_macos_conda:
    <<: *binary_common
    macos:
452
      xcode: "12.0"
453
    steps:
454
      - checkout_merge
455
      - designate_upload_channel
456
457
458
459
460
461
462
463
464
      - 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
465
466
467
468
      - persist_to_workspace:
          root: /Users/distiller/miniconda3/conda-bld/osx-64
          paths:
            - "*"
469
470
      - store_test_results:
          path: build_results/
Edward Z. Yang's avatar
Edward Z. Yang committed
471
472
473
474
475
476
477
478

  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
479
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
480
481
482
483
484
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
485
            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
486
487
488

  # Requires org-member context
  binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
489
490
491
492
    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
493
494
495
496
497
    docker:
      - image: circleci/python:3.7
    steps:
      - attach_workspace:
          at: ~/workspace
498
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
499
500
501
502
503
504
505
506
507
508
      - 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
509
            for pkg in ~/workspace/*.whl; do
510
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
511
            done
512

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

620
621
622
623
624
625
626
  unittest_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout
627
      - designate_upload_channel
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
659
660
661
      - 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:
662
      image: ubuntu-1604-cuda-10.2:202012-01
663
    resource_class: gpu.nvidia.medium
664
    environment:
665
      image_name: "pytorch/manylinux-cuda102"
666
      PYTHON_VERSION: << parameters.python_version >>
667
668
    steps:
      - checkout
669
      - designate_upload_channel
670
671
672
673
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
674
675
676
677
678
      - restore_cache:

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

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

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

          paths:
            - conda
            - env
689
690
691
692
693
694
695
696
      - run:
          # Here we create an envlist file that contains some env variables that we want the docker container to be aware of.
          # Normally, the CIRCLECI variable is set and available on all CI workflows: https://circleci.com/docs/2.0/env-vars/#built-in-environment-variables.
          # They're avaiable in all the other workflows (OSX and Windows).
          # But here, we're running the unittest_linux_gpu workflows in a docker container, where those variables aren't accessible.
          # So instead we dump the variables we need in env.list and we pass that file when invoking "docker run".
          name: export CIRCLECI env var
          command: echo "CIRCLECI=true" >> ./env.list
697
698
      - run:
          name: Install torchvision
699
          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
700
701
      - run:
          name: Run tests
702
          command: docker run --env-file ./env.list -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
703
704
705
706
707
708
709
710
711
712
713
714
      - 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
715
      - designate_upload_channel
716
      - install_cuda_compatible_cmath
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
742
743
744
745
746
747
748
749
750
751
752
      - 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:
753
      CUDA_VERSION: "10.2"
754
      PYTHON_VERSION: << parameters.python_version >>
755
756
    steps:
      - checkout
757
      - designate_upload_channel
758
      - install_cuda_compatible_cmath
759
760
761
762
      - run:
          name: Generate cache key
          # This will refresh cache on Sundays, nightly build should generate new cache.
          command: echo "$(date +"%Y-%U")" > .circleci-weekly
763
764
765
766
      - restore_cache:

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

768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
      - 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
789

Francisco Massa's avatar
Francisco Massa committed
790
791
792
  unittest_macos_cpu:
    <<: *binary_common
    macos:
793
      xcode: "12.0"
Francisco Massa's avatar
Francisco Massa committed
794
795
796
    resource_class: large
    steps:
      - checkout
797
      - designate_upload_channel
Francisco Massa's avatar
Francisco Massa committed
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
      - 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

833
834
835
836
837
838
839
  cmake_linux_cpu:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda102"
    resource_class: 2xlarge+
    steps:
      - checkout_merge
840
      - designate_upload_channel
841
842
843
844
845
846
847
848
      - run:
          name: Setup conda
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run: packaging/build_cmake.sh

  cmake_linux_gpu:
    <<: *binary_common
    machine:
849
      image: ubuntu-1604-cuda-10.2:202012-01
850
851
852
853
854
855
856
857
    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
858
      - designate_upload_channel
859
860
861
862
863
      - 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
864
          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
865
866
867
868

  cmake_macos_cpu:
    <<: *binary_common
    macos:
869
      xcode: "12.0"
870
871
    steps:
      - checkout_merge
872
      - designate_upload_channel
873
874
875
876
877
878
879
880
881
882
883
884
885
886
      - 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
887
      - designate_upload_channel
888
      - install_cuda_compatible_cmath
889
890
891
892
893
894
895
896
897
898
899
900
      - 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
901
      - designate_upload_channel
902
      - install_cuda_compatible_cmath
903
904
905
906
907
908
909
      - run:
          command: |
            set -ex
            source packaging/windows/internal/vc_install_helper.sh
            packaging/windows/internal/cuda_install.bat
            packaging/build_cmake.sh

910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
  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
926
927
928
929
930
931
932
933
934
      - 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" }}

935
936
937
938
939
      - run:
          name: Build docs
          command: |
            set -ex
            tag=${CIRCLE_TAG:1:5}
940
            VERSION=${tag:-main}
941
942
943
944
945
946
            eval "$(./conda/bin/conda shell.bash hook)"
            conda activate ./env
            pushd docs
            pip install -r requirements.txt
            make html
            popd
947
948
949
950
951
952
      - save_cache:

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

          paths:
            - ./docs/source/auto_examples
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
979
980
981
982
983
984
985
      - 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}
986
            target=${tag:-main}
987
988
989
            ~/workspace/.circleci/build_docs/commit_docs.sh ~/workspace $target


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

1590
1591
1592
1593
1594
1595
1596
1597
1598
1599
1600
1601
1602
1603
  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'
1604
1605
1606
1607
      - unittest_linux_cpu:
          cu_version: cpu
          name: unittest_linux_cpu_py3.9
          python_version: '3.9'
1608
      - unittest_linux_gpu:
1609
          cu_version: cu102
1610
1611
1612
          filters:
            branches:
              only:
1613
              - main
1614
1615
1616
1617
              - nightly
          name: unittest_linux_gpu_py3.6
          python_version: '3.6'
      - unittest_linux_gpu:
1618
          cu_version: cu102
1619
1620
1621
          filters:
            branches:
              only:
1622
              - main
1623
1624
1625
1626
              - nightly
          name: unittest_linux_gpu_py3.7
          python_version: '3.7'
      - unittest_linux_gpu:
1627
          cu_version: cu102
1628
1629
          name: unittest_linux_gpu_py3.8
          python_version: '3.8'
1630
      - unittest_linux_gpu:
1631
          cu_version: cu102
1632
1633
1634
          filters:
            branches:
              only:
1635
              - main
1636
1637
1638
              - nightly
          name: unittest_linux_gpu_py3.9
          python_version: '3.9'
1639
1640
1641
1642
1643
1644
1645
1646
1647
1648
1649
1650
      - 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'
1651
1652
1653
1654
      - unittest_windows_cpu:
          cu_version: cpu
          name: unittest_windows_cpu_py3.9
          python_version: '3.9'
1655
      - unittest_windows_gpu:
1656
          cu_version: cu102
1657
1658
1659
          filters:
            branches:
              only:
1660
              - main
1661
1662
1663
1664
              - nightly
          name: unittest_windows_gpu_py3.6
          python_version: '3.6'
      - unittest_windows_gpu:
1665
          cu_version: cu102
1666
1667
1668
          filters:
            branches:
              only:
1669
              - main
1670
1671
1672
1673
              - nightly
          name: unittest_windows_gpu_py3.7
          python_version: '3.7'
      - unittest_windows_gpu:
1674
          cu_version: cu102
1675
1676
          name: unittest_windows_gpu_py3.8
          python_version: '3.8'
1677
      - unittest_windows_gpu:
1678
          cu_version: cu102
1679
1680
1681
          filters:
            branches:
              only:
1682
              - main
1683
1684
1685
              - nightly
          name: unittest_windows_gpu_py3.9
          python_version: '3.9'
Francisco Massa's avatar
Francisco Massa committed
1686
1687
1688
1689
1690
1691
1692
1693
1694
1695
1696
1697
      - 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'
1698
1699
1700
1701
      - unittest_macos_cpu:
          cu_version: cpu
          name: unittest_macos_cpu_py3.9
          python_version: '3.9'
1702
1703
1704
1705
1706
1707
1708
1709

  cmake:
    jobs:
      - cmake_linux_cpu:
          cu_version: cpu
          name: cmake_linux_cpu
          python_version: '3.8'
      - cmake_linux_gpu:
1710
          cu_version: cu102
1711
1712
          name: cmake_linux_gpu
          python_version: '3.8'
1713
          wheel_docker_image: pytorch/manylinux-cuda102
1714
1715
1716
1717
      - cmake_windows_cpu:
          cu_version: cpu
          name: cmake_windows_cpu
          python_version: '3.8'
1718
      - cmake_windows_gpu:
1719
          cu_version: cu102
1720
1721
          name: cmake_windows_gpu
          python_version: '3.8'
1722
1723
1724
1725
1726
      - cmake_macos_cpu:
          cu_version: cpu
          name: cmake_macos_cpu
          python_version: '3.8'

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