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

9
10
11
12
13
14
executors:
  windows-cpu:
    machine:
      resource_class: windows.xlarge
      image: windows-server-2019-vs2019:stable
      shell: bash.exe
peterjc123's avatar
peterjc123 committed
15

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

22
commands:
23
24
25
26
27
28
  generate_cache_key:
    description: "Generates a cache key file that changes daily"
    steps:
      - run:
          name: Generate cache key
          command: echo "$(date +"%Y-%m-%d")" > .cachekey
29
30
31
32
33
34
  designate_upload_channel:
    description: "inserts the correct upload channel into ${BASH_ENV}"
    steps:
      - run:
          name: adding UPLOAD_CHANNEL to BASH_ENV
          command: |
35
36
            # Hardcoded for release branch
            echo "export UPLOAD_CHANNEL=test" >> ${BASH_ENV}
moto's avatar
moto committed
37
38
  install_build_tools_macos:
    description: "installs tools required to build torchaudio"
moto's avatar
moto committed
39
40
    steps:
      - run:
41
          name: Install build tools
moto's avatar
moto committed
42
          command: HOMEBREW_NO_AUTO_UPDATE=1 brew install pkg-config wget
moto's avatar
moto committed
43
          # Disable brew auto update which is very slow
Eli Uriegas's avatar
Eli Uriegas committed
44
45
46
47
48
49
50
  load_conda_channel_flags:
    description: "Determines whether we need extra conda channels"
    steps:
      - run:
          name: Adding CONDA_CHANNEL_FLAGS to BASH_ENV
          command: |
              CONDA_CHANNEL_FLAGS=""
51
              # formerly used to add conda-forge flags for Python 3.9, reserving the mechanism for future python upgrades
52
53
54
55
56
57
58
  windows_install_cuda:
    description: "Install desired CUDA version on Windows runners"
    steps:
      - run:
          name: Install CUDA
          command: |
              packaging/windows/internal/cuda_install.bat
59

60
61
binary_common: &binary_common
  parameters:
Edward Z. Yang's avatar
Edward Z. Yang committed
62
    # Edit these defaults to do a release
63
64
65
    build_version:
      description: "version number of release binary; by default, build a nightly"
      type: string
66
      default: "0.10.0"
67
68
69
    pytorch_version:
      description: "PyTorch version to build against; by default, use a nightly"
      type: string
70
      default: "1.10.0"
71
72
    # Don't edit these
    python_version:
73
      description: "Python version to build against (e.g., 3.8)"
74
      type: string
Nikita Shulga's avatar
Nikita Shulga committed
75
76
77
78
79
80
81
82
83
84
85
86
    cuda_version:
      description: "CUDA version to build against (e.g., cpu, cu101)"
      type: string
      default: "cpu"
    wheel_docker_image:
      description: "Wheel only: what docker image to use"
      type: string
      default: "pytorch/manylinux-cuda102"
    conda_docker_image:
      description: "Conda only: what docker image to use"
      type: string
      default: "pytorch/conda-builder:cuda102"
87
  environment: &environment
88
89
90
    PYTHON_VERSION: << parameters.python_version >>
    BUILD_VERSION: << parameters.build_version >>
    PYTORCH_VERSION: << parameters.pytorch_version >>
Nikita Shulga's avatar
Nikita Shulga committed
91
    CU_VERSION: << parameters.cuda_version >>
92

Mingbo Wan's avatar
Mingbo Wan committed
93
94
95
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
Caroline Chen's avatar
Caroline Chen committed
96
97
    - image: pytorch/torchaudio_unittest_base:smoke_test-20211019
  resource_class: large
Mingbo Wan's avatar
Mingbo Wan committed
98

99
100
101
jobs:
  circleci_consistency:
    docker:
moto's avatar
moto committed
102
      - image: cimg/python:3.8
103
104
105
106
    steps:
      - checkout
      - run:
          command: |
107
            pip install --user --progress-bar off jinja2 pyyaml
108
109
110
            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)

111
112
  download_third_parties_nix:
    docker:
moto's avatar
moto committed
113
      - image: "pytorch/torchaudio_unittest_base:manylinux"
114
115
116
    resource_class: small
    steps:
      - checkout
117
      - generate_cache_key
118
119
120
      - restore_cache:
          {% raw %}
          keys:
moto's avatar
moto committed
121
            - tp-nix-v2-{{ checksum ".cachekey" }}
122
123
          {% endraw %}
      - run:
moto's avatar
moto committed
124
          command: |
125
126
              mkdir -p third_party/sox/archives/
              wget --no-clobber --directory-prefix=third_party/sox/archives/ $(awk '/URL /{print $2}' third_party/sox/CMakeLists.txt)
127
128
      - save_cache:
          {% raw %}
moto's avatar
moto committed
129
          key: tp-nix-v2-{{ checksum ".cachekey" }}
130
131
          {% endraw %}
          paths:
132
            - third_party/sox/archives
133
134
135
      - persist_to_workspace:
          root: third_party
          paths:
136
            - sox/archives
137

138
139
140
  binary_linux_wheel:
    <<: *binary_common
    docker:
Nikita Shulga's avatar
Nikita Shulga committed
141
      - image: << parameters.wheel_docker_image >>
142
143
144
    resource_class: 2xlarge+
    steps:
      - checkout
145
      - designate_upload_channel
146
147
      - attach_workspace:
          at: third_party
148
149
150
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
151
152
153
154
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
155
156
157
158

  binary_linux_conda:
    <<: *binary_common
    docker:
Nikita Shulga's avatar
Nikita Shulga committed
159
      - image: "<< parameters.conda_docker_image >>"
160
161
162
    resource_class: 2xlarge+
    steps:
      - checkout
Eli Uriegas's avatar
Eli Uriegas committed
163
      - load_conda_channel_flags
164
165
      - attach_workspace:
          at: third_party
166
167
168
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
169
      - persist_to_workspace:
170
          root: /opt/conda
Edward Z. Yang's avatar
Edward Z. Yang committed
171
          paths:
172
            - "conda-bld/*"
173
174
175
176

  binary_macos_wheel:
    <<: *binary_common
    macos:
177
      xcode: "12.0"
178
179
    steps:
      - checkout
moto's avatar
moto committed
180
      - install_build_tools_macos
181
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
182
      - load_conda_channel_flags
183
184
      - attach_workspace:
          at: third_party
185
186
187
188
189
190
191
192
193
194
195
      - 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
196
197
198
199
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
200
201
202
203

  binary_macos_conda:
    <<: *binary_common
    macos:
204
      xcode: "12.0"
205
206
    steps:
      - checkout
moto's avatar
moto committed
207
      - install_build_tools_macos
Eli Uriegas's avatar
Eli Uriegas committed
208
      - load_conda_channel_flags
209
210
      - attach_workspace:
          at: third_party
211
212
213
214
215
216
217
218
219
      - 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
220
      - persist_to_workspace:
221
          root: /Users/distiller/miniconda3
Edward Z. Yang's avatar
Edward Z. Yang committed
222
          paths:
223
            - "conda-bld/*"
224

peterjc123's avatar
peterjc123 committed
225
226
227
228
229
230
  binary_windows_wheel:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout
231
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
232
      - load_conda_channel_flags
233
      - windows_install_cuda
peterjc123's avatar
peterjc123 committed
234
      - run:
235
          name: Build wheel packages
peterjc123's avatar
peterjc123 committed
236
          command: |
237
            set -ex
peterjc123's avatar
peterjc123 committed
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            bash packaging/build_wheel.sh
      - store_artifacts:
          path: dist
      - persist_to_workspace:
          root: dist
          paths:
            - "*"

  binary_windows_conda:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout
Eli Uriegas's avatar
Eli Uriegas committed
254
      - load_conda_channel_flags
255
      - windows_install_cuda
peterjc123's avatar
peterjc123 committed
256
      - run:
257
          name: Build conda packages
258
          no_output_timeout: 20m
peterjc123's avatar
peterjc123 committed
259
          command: |
260
            set -ex
peterjc123's avatar
peterjc123 committed
261
262
263
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
264
265
266
267
            # cudatoolkit >= 11 isn't available for windows in the nvidia channel
            if [[ "${CU_VERSION}" =~ cu11.* ]]; then
              export CONDA_CHANNEL_FLAGS="-c conda-forge"
            fi
peterjc123's avatar
peterjc123 committed
268
269
270
271
            bash packaging/build_conda.sh
      - store_artifacts:
          path: C:/tools/miniconda3/conda-bld/win-64
      - persist_to_workspace:
272
          root: C:/tools/miniconda3
peterjc123's avatar
peterjc123 committed
273
          paths:
274
            - "conda-bld/*"
peterjc123's avatar
peterjc123 committed
275

Edward Z. Yang's avatar
Edward Z. Yang committed
276
277
278
279
280
281
282
  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
283
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
284
285
286
287
288
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
289
            anaconda -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/conda-bld/*/*.tar.bz2 -u "pytorch-${UPLOAD_CHANNEL}" --label main --no-progress --force
Edward Z. Yang's avatar
Edward Z. Yang committed
290
291
292

  # Requires org-member context
  binary_wheel_upload:
293
294
295
296
    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
297
    docker:
moto's avatar
moto committed
298
      - image: cimg/python:3.8
Edward Z. Yang's avatar
Edward Z. Yang committed
299
300
301
302
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
303
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
304
305
306
307
308
309
310
311
312
313
      - 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
            for pkg in ~/workspace/*.whl; do
314
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
315
316
            done

Mingbo Wan's avatar
Mingbo Wan committed
317
318
319
  smoke_test_linux_conda:
    <<: *smoke_test_common
    steps:
Mingbo Wan's avatar
Mingbo Wan committed
320
321
      - attach_workspace:
          at: ~/workspace
322
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
323
      - load_conda_channel_flags
Mingbo Wan's avatar
Mingbo Wan committed
324
      - run:
Mingbo Wan's avatar
Mingbo Wan committed
325
          name: install binaries
Mingbo Wan's avatar
Mingbo Wan committed
326
327
328
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
329
            conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
330
            conda install -v -y -c file://$HOME/workspace/conda-bld torchaudio
Mingbo Wan's avatar
Mingbo Wan committed
331
332
333
334
      - run:
          name: smoke test
          command: |
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
Mingbo Wan's avatar
Mingbo Wan committed
335
336
            python -c "import torchaudio"

Caroline Chen's avatar
Caroline Chen committed
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
  smoke_test_linux_conda_gpu:
    <<: *smoke_test_common
    steps:
      - attach_workspace:
          at: ~/workspace
      - designate_upload_channel
      - load_conda_channel_flags
      - 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-${UPLOAD_CHANNEL} pytorch cudatoolkit=${CU_VERSION:2:2}.${CU_VERSION:4} -c conda-forge
            conda install -v -y -c file://$HOME/workspace/conda-bld torchaudio
      - run:
          name: smoke test
          command: |
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
            python -c "import torchaudio"

Mingbo Wan's avatar
Mingbo Wan committed
357
358
359
  smoke_test_linux_pip:
    <<: *smoke_test_common
    steps:
Mingbo Wan's avatar
Mingbo Wan committed
360
361
      - attach_workspace:
          at: ~/workspace
362
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
363
      - load_conda_channel_flags
Mingbo Wan's avatar
Mingbo Wan committed
364
      - run:
Mingbo Wan's avatar
Mingbo Wan committed
365
          name: install binaries
Mingbo Wan's avatar
Mingbo Wan committed
366
367
368
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
Caroline Chen's avatar
Caroline Chen committed
369
            pip install $(ls ~/workspace/torchaudio*.whl) -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${CU_VERSION}/torch_${UPLOAD_CHANNEL}.html"
Mingbo Wan's avatar
Mingbo Wan committed
370
371
372
373
      - run:
          name: smoke test
          command: |
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
Mingbo Wan's avatar
Mingbo Wan committed
374
375
            python -c "import torchaudio"

376
377
378
379
380
381
382
  smoke_test_windows_conda:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - attach_workspace:
          at: ~/workspace
383
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
384
      - load_conda_channel_flags
385
386
387
388
389
390
391
392
      - 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
            conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
            conda activate python${PYTHON_VERSION}
393
            conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
            conda install -v -y $(ls ~/workspace/torchaudio*.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 torchaudio"

  smoke_test_windows_pip:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - attach_workspace:
          at: ~/workspace
409
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
410
      - load_conda_channel_flags
411
412
413
414
415
416
417
418
      - 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
            conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
            conda activate python${PYTHON_VERSION}
419
            pip install $(ls ~/workspace/torchaudio*.whl) -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/torch_${UPLOAD_CHANNEL}.html"
420
421
422
423
424
425
426
      - run:
          name: smoke test
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate python${PYTHON_VERSION}
            python -c "import torchaudio"

427
  smoke_test_docker_image_build:
Mingbo Wan's avatar
Mingbo Wan committed
428
429
430
431
432
433
434
435
436
437
438
    machine:
      image: ubuntu-1604:201903-01
    resource_class: large
    environment:
      image_name: torchaudio/smoke_test
    steps:
      - checkout
      - run:
          name: build_docker image
          no_output_timeout: "1h"
          command: |
439
            cd .circleci/smoke_test/docker && docker build . -t ${image_name}:${CIRCLE_WORKFLOW_ID}
Mingbo Wan's avatar
Mingbo Wan committed
440
441
442
443
444
445
446
447
448
449
      - run:
          name: upload docker image
          no_output_timeout: "1h"
          command: |
            set +x
            export AWS_ACCESS_KEY_ID=${ECR_AWS_ACCESS_KEY}
            export AWS_SECRET_ACCESS_KEY=${ECR_AWS_SECRET_ACCESS_KEY}
            eval $(aws ecr get-login --region us-east-1 --no-include-email)
            set -x
            docker tag ${image_name}:${CIRCLE_WORKFLOW_ID} 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:${CIRCLE_WORKFLOW_ID}
450
            docker tag ${image_name}:${CIRCLE_WORKFLOW_ID} 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:latest
Mingbo Wan's avatar
Mingbo Wan committed
451
            docker push 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:${CIRCLE_WORKFLOW_ID}
452
            docker push 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:latest
Edward Z. Yang's avatar
Edward Z. Yang committed
453

moto's avatar
moto committed
454
  unittest_linux_cpu:
moto's avatar
moto committed
455
456
    <<: *binary_common
    docker:
457
      - image: pytorch/torchaudio_unittest_base:manylinux-20210121
moto's avatar
moto committed
458
459
460
    resource_class: 2xlarge+
    steps:
      - checkout
461
462
      - attach_workspace:
          at: third_party
463
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
464
      - load_conda_channel_flags
465
466
467
468
469
470
471
472
473
474
475
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run:
          name: Install torchaudio
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/linux/scripts/run_test.sh
      - store_test_results:
          path: test-results
476
477
      - store_artifacts:
          path: test/htmlcov
moto's avatar
moto committed
478
479
480
481
482
483
  unittest_linux_gpu:
    <<: *binary_common
    machine:
      image: ubuntu-1604-cuda-10.1:201909-23
    resource_class: gpu.small
    environment:
484
      <<: *environment
moto's avatar
moto committed
485
      image_name: pytorch/torchaudio_unittest_base:manylinux-cuda10.2-cudnn8-20210623
moto's avatar
moto committed
486
487
    steps:
      - checkout
488
489
      - attach_workspace:
          at: third_party
490
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
491
      - load_conda_channel_flags
492
493
494
      - run:
          name: Pull Docker image
          command: docker pull --quiet "${image_name}"
moto's avatar
moto committed
495
496
      - run:
          name: Setup
497
          command: docker run -t --gpus all -e PYTHON_VERSION -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/setup_env.sh
moto's avatar
moto committed
498
499
      - run:
          name: Install torchaudio
Eli Uriegas's avatar
Eli Uriegas committed
500
          command: docker run -t --gpus all -e UPLOAD_CHANNEL -e CONDA_CHANNEL_FLAGS -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh
moto's avatar
moto committed
501
502
      - run:
          name: Run tests
503
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e "TORCHAUDIO_TEST_FORCE_CUDA=1" "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
moto's avatar
moto committed
504
505
      - store_test_results:
          path: test-results
506
507
      - store_artifacts:
          path: test/htmlcov
moto's avatar
moto committed
508
509

  unittest_windows_cpu:
510
511
512
    <<: *binary_common
    executor:
      name: windows-cpu
513
514
    steps:
      - checkout
515
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
516
      - load_conda_channel_flags
517
518
519
520
521
522
523
524
525
526
527
      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - run:
          name: Install torchaudio
          command: .circleci/unittest/windows/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/windows/scripts/run_test.sh
      - store_test_results:
          path: test-results
528
529
      - store_artifacts:
          path: test/htmlcov
530
531
532
533
534
535

  unittest_windows_gpu:
    <<: *binary_common
    executor:
      name: windows-gpu
    environment:
536
      <<: *environment
537
      CUDA_VERSION: "10.2"
538
      TORCHAUDIO_TEST_FORCE_CUDA: 1
539
540
    steps:
      - checkout
541
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
542
      - load_conda_channel_flags
moto's avatar
moto committed
543
544
      - run:
          name: Setup
545
          command: .circleci/unittest/windows/scripts/setup_env.sh
546
547
548
549
550
551
      - run:
          name: Install CUDA
          command: packaging/windows/internal/cuda_install.bat
      - run:
          name: Update CUDA driver
          command: packaging/windows/internal/driver_update.bat
moto's avatar
moto committed
552
      - run:
553
          name: Install torchaudio
554
          command: .circleci/unittest/windows/scripts/install.sh
moto's avatar
moto committed
555
      - run:
556
          name: Run tests
557
          command: .circleci/unittest/windows/scripts/run_test.sh
moto's avatar
moto committed
558
559
      - store_test_results:
          path: test-results
560
561
      - store_artifacts:
          path: test/htmlcov
moto's avatar
moto committed
562

moto's avatar
moto committed
563
564
565
  unittest_macos_cpu:
    <<: *binary_common
    macos:
566
      xcode: "12.0"
moto's avatar
moto committed
567
568
569
570
    resource_class: large
    steps:
      - checkout
      - install_build_tools_macos
Eli Uriegas's avatar
Eli Uriegas committed
571
      - load_conda_channel_flags
moto's avatar
moto committed
572
573
      - attach_workspace:
          at: third_party
574
      - designate_upload_channel
moto's avatar
moto committed
575
576
577
578
579
580
581
582
583
584
585
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run:
          name: Install torchaudio
          command: .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Run tests
          command: .circleci/unittest/linux/scripts/run_test.sh
      - store_test_results:
          path: test-results
586
587
      - store_artifacts:
          path: test/htmlcov
moto's avatar
moto committed
588

589
590
591
592
593
594
595
  stylecheck:
    <<: *binary_common
    docker:
      - image: "pytorch/torchaudio_unittest_base:manylinux"
    resource_class: medium
    steps:
      - checkout
596
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
597
      - load_conda_channel_flags
598
599
600
601
602
603
604
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - run:
          name: Run style check
          command: .circleci/unittest/linux/scripts/run_style_checks.sh

605
606
607
608
609
610
611
612
613
  build_docs:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda100"
    resource_class: 2xlarge+
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
614
      - designate_upload_channel
Eli Uriegas's avatar
Eli Uriegas committed
615
      - load_conda_channel_flags
616
617
618
619
620
621
622
      - run:
          name: Install pytorch-audio
          command: .circleci/build_docs/install_wheels.sh
      - run:
          name: Build docs
          command: .circleci/build_docs/build_docs.sh
      - persist_to_workspace:
623
          root: ./
624
625
          paths:
            - "*"
626
627
628
      - store_artifacts:
          path: ./docs/build/html
          destination: docs
629
630
631
632
633
634
635
636
637

  upload_docs:
    <<: *binary_common
    docker:
      - image: "pytorch/manylinux-cuda100"
    resource_class: 2xlarge+
    steps:
      - attach_workspace:
          at: ~/workspace
638
639
640
641
642
643
644
645
646
647
      - 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
648
649
650
      - run:
          name: Upload docs
          command: |
651
652
            # Don't use "checkout" step since it uses ssh, which cannot git push
            # https://circleci.com/docs/2.0/configuration-reference/#checkout
653
654
            set -ex
            tag=${CIRCLE_TAG:1:5}
655
            target=${tag:-main}
Matti Picus's avatar
Matti Picus committed
656
            ~/workspace/.circleci/build_docs/commit_docs.sh ~/workspace $target
657

658
659
660
  docstring_parameters_sync:
    <<: *binary_common
    docker:
moto's avatar
moto committed
661
      - image: cimg/python:3.8
662
663
664
665
666
667
668
669
    steps:
      - checkout
      - run:
          name: Check parameters docstring sync
          command: |
            pip install --user pydocstyle
            pydocstyle torchaudio

Edward Z. Yang's avatar
Edward Z. Yang committed
670
671
672
673
workflows:
  build:
    jobs:
      - circleci_consistency
moto's avatar
moto committed
674
675
676
677
      {{ build_workflows() }}
  unittest:
    jobs:
      {{ unittest_workflows() }}
Edward Z. Yang's avatar
Edward Z. Yang committed
678
679
  nightly:
    jobs:
moto's avatar
moto committed
680
681
682
683
      - circleci_consistency:
          filters:
            branches:
              only: nightly
moto's avatar
moto committed
684
      {{ build_workflows(prefix="nightly_", filter_branch="nightly", upload=True) }}
Mingbo Wan's avatar
Mingbo Wan committed
685
686
687
688
689
690
691
  docker_build:
    triggers:
      - schedule:
          cron: "0 10 * * 0"
          filters:
            branches:
              only:
692
                - main
Mingbo Wan's avatar
Mingbo Wan committed
693
    jobs:
694
      - smoke_test_docker_image_build