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

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

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

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

22
23
24
25
26
commands:
  checkout_merge:
    description: "checkout merge branch"
    steps:
      - checkout
27
28
29
30
31
32
33
34
35
#     - run:
#         name: Checkout merge branch
#         command: |
#           set -ex
#           BRANCH=$(git rev-parse --abbrev-ref HEAD)
#           if [[ "$BRANCH" != "master" ]]; then
#             git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
#             git checkout "merged/$CIRCLE_BRANCH"
#           fi
36
37
38
39
40
41
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-cuda101"
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
195
  python_lint:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
            pip install --user --progress-bar off flake8 typing
196
197
198
199
200
201
202
203
204
            flake8 --config=setup.cfg .

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

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

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

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

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

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

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

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

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

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

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

405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
  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"

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

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

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

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

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

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

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

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

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

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

761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
      - 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
782

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

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

  cmake_linux_gpu:
    <<: *binary_common
    machine:
      image: ubuntu-1604-cuda-10.1:201909-23
    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
851
      - designate_upload_channel
852
853
854
855
856
      - 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
857
          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
858
859
860
861

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

903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
  build_docs:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda100"
    resource_class: 2xlarge+
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - designate_upload_channel
      - run:
          name: Install torchvision
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Build docs
          command: |
            set -ex
            tag=${CIRCLE_TAG:1:5}
            VERSION=${tag:-master}
            eval "$(./conda/bin/conda shell.bash hook)"
            conda activate ./env
            pushd docs
            pip install -r requirements.txt
            make html
            popd
      - persist_to_workspace:
          root: ./
          paths:
            - "*"
      - store_artifacts:
          path: ./docs/build/html
          destination: docs

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


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

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

  cmake:
    jobs:
      - cmake_linux_cpu:
          cu_version: cpu
          name: cmake_linux_cpu
          python_version: '3.8'
      - cmake_linux_gpu:
          cu_version: cu101
          name: cmake_linux_gpu
          python_version: '3.8'
          wheel_docker_image: pytorch/manylinux-cuda101
      - cmake_windows_cpu:
          cu_version: cpu
          name: cmake_windows_cpu
          python_version: '3.8'
1565
1566
1567
1568
      - cmake_windows_gpu:
          cu_version: cu101
          name: cmake_windows_gpu
          python_version: '3.8'
1569
1570
1571
1572
1573
      - cmake_macos_cpu:
          cu_version: cpu
          name: cmake_macos_cpu
          python_version: '3.8'

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