config.yml.in 22.2 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
19
20
21
  windows-gpu:
    machine:
      resource_class: windows.gpu.nvidia.medium
      image: windows-server-2019-nvidia:stable
      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
35
36
  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
37
            if [[ -n "${CIRCLE_TAG}" ]] || [[ ${CIRCLE_BRANCH} =~ release/* ]]; then
38
39
40
              our_upload_channel=test
            fi
            echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
moto's avatar
moto committed
41
42
  install_build_tools_macos:
    description: "installs tools required to build torchaudio"
moto's avatar
moto committed
43
44
    steps:
      - run:
moto's avatar
moto committed
45
          name: Install cmake and pkg-config
moto's avatar
moto committed
46
          command: HOMEBREW_NO_AUTO_UPDATE=1 brew install cmake pkg-config wget
moto's avatar
moto committed
47
          # Disable brew auto update which is very slow
48

49
50
binary_common: &binary_common
  parameters:
Edward Z. Yang's avatar
Edward Z. Yang committed
51
    # Edit these defaults to do a release
52
53
54
55
56
57
58
59
60
61
    build_version:
      description: "version number of release binary; by default, build a nightly"
      type: string
      default: ""
    pytorch_version:
      description: "PyTorch version to build against; by default, use a nightly"
      type: string
      default: ""
    # Don't edit these
    python_version:
62
      description: "Python version to build against (e.g., 3.8)"
63
64
65
66
67
      type: string
  environment:
    PYTHON_VERSION: << parameters.python_version >>
    BUILD_VERSION: << parameters.build_version >>
    PYTORCH_VERSION: << parameters.pytorch_version >>
Edward Z. Yang's avatar
Edward Z. Yang committed
68
    CU_VERSION: cpu
69

Mingbo Wan's avatar
Mingbo Wan committed
70
71
72
smoke_test_common: &smoke_test_common
  <<: *binary_common
  docker:
73
    - image: 308535385114.dkr.ecr.us-east-1.amazonaws.com/torchaudio/smoke_test:56c846a5-acaa-41a7-92f5-46ec66186c61
Mingbo Wan's avatar
Mingbo Wan committed
74
75
76
77
      aws_auth:
        aws_access_key_id: ${ECR_AWS_ACCESS_KEY}
        aws_secret_access_key: ${ECR_AWS_SECRET_ACCESS_KEY}

78
79
80
jobs:
  circleci_consistency:
    docker:
81
      - image: circleci/python:3.8
82
83
84
85
    steps:
      - checkout
      - run:
          command: |
86
            pip install --user --progress-bar off jinja2 pyyaml
87
88
89
            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)

90
91
  download_third_parties_nix:
    docker:
moto's avatar
moto committed
92
      - image: "pytorch/torchaudio_unittest_base:manylinux"
93
94
95
    resource_class: small
    steps:
      - checkout
96
      - generate_cache_key
97
98
99
      - restore_cache:
          {% raw %}
          keys:
moto's avatar
moto committed
100
            - tp-nix-v2-{{ checksum ".cachekey" }}
101
102
          {% endraw %}
      - run:
moto's avatar
moto committed
103
104
105
          command: |
              mkdir -p third_party/archives/
              wget --no-clobber --directory-prefix=third_party/archives/ $(awk '/URL /{print $2}' third_party/CMakeLists.txt)
106
107
      - save_cache:
          {% raw %}
moto's avatar
moto committed
108
          key: tp-nix-v2-{{ checksum ".cachekey" }}
109
110
          {% endraw %}
          paths:
moto's avatar
moto committed
111
            - third_party/archives
112
113
114
      - persist_to_workspace:
          root: third_party
          paths:
moto's avatar
moto committed
115
            - archives
116

117
118
119
  binary_linux_wheel:
    <<: *binary_common
    docker:
120
      - image: "pytorch/manylinux-cuda100"
121
122
123
    resource_class: 2xlarge+
    steps:
      - checkout
124
125
      - attach_workspace:
          at: third_party
126
127
128
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
129
130
131
132
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
133
134
135
136

  binary_linux_conda:
    <<: *binary_common
    docker:
137
      - image: "pytorch/conda-cuda"
138
139
140
    resource_class: 2xlarge+
    steps:
      - checkout
141
142
      - attach_workspace:
          at: third_party
143
144
145
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
146
      - persist_to_workspace:
147
          root: /opt/conda
Edward Z. Yang's avatar
Edward Z. Yang committed
148
          paths:
149
            - "conda-bld/*"
150
151
152
153

  binary_macos_wheel:
    <<: *binary_common
    macos:
154
      xcode: "9.4.1"
155
156
    steps:
      - checkout
moto's avatar
moto committed
157
      - install_build_tools_macos
158
159
      - attach_workspace:
          at: third_party
160
161
162
163
164
165
166
167
168
169
170
      - 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
171
172
173
174
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
175
176
177
178

  binary_macos_conda:
    <<: *binary_common
    macos:
179
      xcode: "9.4.1"
180
181
    steps:
      - checkout
moto's avatar
moto committed
182
      - install_build_tools_macos
183
184
      - attach_workspace:
          at: third_party
185
186
187
188
189
190
191
192
193
      - 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
194
      - persist_to_workspace:
195
          root: /Users/distiller/miniconda3
Edward Z. Yang's avatar
Edward Z. Yang committed
196
          paths:
197
            - "conda-bld/*"
198

peterjc123's avatar
peterjc123 committed
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
  binary_windows_wheel:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - checkout
      - run:
          name: build
          command: |
            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
      - run:
          name: build
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
            bash packaging/build_conda.sh
      - store_artifacts:
          path: C:/tools/miniconda3/conda-bld/win-64
      - persist_to_workspace:
234
          root: C:/tools/miniconda3
peterjc123's avatar
peterjc123 committed
235
          paths:
236
            - "conda-bld/*"
peterjc123's avatar
peterjc123 committed
237

Edward Z. Yang's avatar
Edward Z. Yang committed
238
239
240
241
242
243
244
  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
245
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
246
247
248
249
250
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set -x
251
            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
252
253
254
255

  # Requires org-member context
  binary_wheel_upload:
    docker:
256
      - image: circleci/python:3.8
Edward Z. Yang's avatar
Edward Z. Yang committed
257
258
259
260
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
261
      - designate_upload_channel
Edward Z. Yang's avatar
Edward Z. Yang committed
262
263
264
265
266
267
268
269
270
271
      - 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
272
              aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/" --acl public-read
Edward Z. Yang's avatar
Edward Z. Yang committed
273
274
            done

Mingbo Wan's avatar
Mingbo Wan committed
275
276
277
  smoke_test_linux_conda:
    <<: *smoke_test_common
    steps:
Mingbo Wan's avatar
Mingbo Wan committed
278
279
      - attach_workspace:
          at: ~/workspace
280
      - designate_upload_channel
Mingbo Wan's avatar
Mingbo Wan committed
281
      - run:
Mingbo Wan's avatar
Mingbo Wan committed
282
          name: install binaries
Mingbo Wan's avatar
Mingbo Wan committed
283
284
285
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
286
            conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
287
            conda install -v -y -c file://$HOME/workspace/conda-bld torchaudio
Mingbo Wan's avatar
Mingbo Wan committed
288
289
290
291
      - 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
292
293
294
295
296
            python -c "import torchaudio"

  smoke_test_linux_pip:
    <<: *smoke_test_common
    steps:
Mingbo Wan's avatar
Mingbo Wan committed
297
298
      - attach_workspace:
          at: ~/workspace
Mingbo Wan's avatar
Mingbo Wan committed
299
      - run:
Mingbo Wan's avatar
Mingbo Wan committed
300
          name: install binaries
Mingbo Wan's avatar
Mingbo Wan committed
301
302
303
          command: |
            set -x
            source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
304
            pip install $(ls ~/workspace/torchaudio*.whl) -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/torch_${UPLOAD_CHANNEL}.html"
Mingbo Wan's avatar
Mingbo Wan committed
305
306
307
308
      - 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
309
310
            python -c "import torchaudio"

311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
  smoke_test_windows_conda:
    <<: *binary_common
    executor:
      name: windows-cpu
    steps:
      - attach_workspace:
          at: ~/workspace
      - 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}
326
            conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
            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
      - 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}
350
            pip install $(ls ~/workspace/torchaudio*.whl) -f "https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/torch_${UPLOAD_CHANNEL}.html"
351
352
353
354
355
356
357
      - run:
          name: smoke test
          command: |
            eval "$('/C/tools/miniconda3/Scripts/conda.exe' 'shell.bash' 'hook')"
            conda activate python${PYTHON_VERSION}
            python -c "import torchaudio"

358
  smoke_test_docker_image_build:
Mingbo Wan's avatar
Mingbo Wan committed
359
360
361
362
363
364
365
366
367
368
369
    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: |
370
            cd .circleci/smoke_test/docker && docker build . -t ${image_name}:${CIRCLE_WORKFLOW_ID}
Mingbo Wan's avatar
Mingbo Wan committed
371
372
373
374
375
376
377
378
379
380
      - 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}
381
            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
382
            docker push 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:${CIRCLE_WORKFLOW_ID}
383
            docker push 308535385114.dkr.ecr.us-east-1.amazonaws.com/${image_name}:latest
Edward Z. Yang's avatar
Edward Z. Yang committed
384

moto's avatar
moto committed
385
  unittest_linux_cpu:
moto's avatar
moto committed
386
387
    <<: *binary_common
    docker:
388
      - image: "pytorch/torchaudio_unittest_base:manylinux"
moto's avatar
moto committed
389
390
391
    resource_class: 2xlarge+
    steps:
      - checkout
392
393
      - attach_workspace:
          at: third_party
394
      - designate_upload_channel
395
      - generate_cache_key
moto's avatar
moto committed
396
397
398
      - restore_cache:
          {% raw %}
          keys:
moto's avatar
moto committed
399
            - env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
400
401
402
403
404
405
          {% endraw %}
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:
          {% raw %}
moto's avatar
moto committed
406
          key: env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
407
408
409
410
          {% endraw %}
          paths:
            - conda
            - env
moto's avatar
moto committed
411
            - third_party/build
moto's avatar
moto committed
412
413
            - third_party/install
            - third_party/src
414
415
416
417
418
419
420
421
422
423
424
425
      - run:
          name: Install torchaudio
          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

moto's avatar
moto committed
426
427
428
429
430
431
432
433
434
  unittest_linux_gpu:
    <<: *binary_common
    machine:
      image: ubuntu-1604-cuda-10.1:201909-23
    resource_class: gpu.small
    environment:
      image_name: "pytorch/torchaudio_unittest_base:manylinux-cuda10.1"
    steps:
      - checkout
435
436
      - attach_workspace:
          at: third_party
437
      - designate_upload_channel
438
      - generate_cache_key
moto's avatar
moto committed
439
440
441
      - restore_cache:
          {% raw %}
          keys:
moto's avatar
moto committed
442
            - env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
moto's avatar
moto committed
443
444
445
446
447
448
          {% endraw %}
      - run:
          name: Setup
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:
          {% raw %}
moto's avatar
moto committed
449
          key: env-v3-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
moto's avatar
moto committed
450
451
452
453
          {% endraw %}
          paths:
            - conda
            - env
moto's avatar
moto committed
454
            - third_party/build
moto's avatar
moto committed
455
456
            - third_party/install
            - third_party/src
moto's avatar
moto committed
457
458
459
460
461
462
463
464
465
466
467
468
469
      - run:
          name: Install torchaudio
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh
      - run:
          name: Run tests
          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
      - 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:
470
471
472
    <<: *binary_common
    executor:
      name: windows-cpu
473
474
    steps:
      - checkout
475
      - designate_upload_channel
476
      - generate_cache_key
477
478
479
      - restore_cache:
          {% raw %}
          keys:
480
            - env-v3-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
481
482
483
484
485
486
          {% endraw %}
      - run:
          name: Setup
          command: .circleci/unittest/windows/scripts/setup_env.sh
      - save_cache:
          {% raw %}
487
          key: env-v3-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
          {% endraw %}
          paths:
            - conda
            - env
      - run:
          name: Install torchaudio
          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"
510
511
    steps:
      - checkout
512
      - designate_upload_channel
513
      - generate_cache_key
514
515
516
      - restore_cache:
          {% raw %}
          keys:
517
            - env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
moto's avatar
moto committed
518
519
520
          {% endraw %}
      - run:
          name: Setup
521
          command: .circleci/unittest/windows/scripts/setup_env.sh
moto's avatar
moto committed
522
523
      - save_cache:
          {% raw %}
524
          key: env-v1-windows-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/windows/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
moto's avatar
moto committed
525
526
527
528
529
          {% endraw %}
          paths:
            - conda
            - env
      - run:
530
          name: Install torchaudio
531
          command: .circleci/unittest/windows/scripts/install.sh
moto's avatar
moto committed
532
      - run:
533
          name: Run tests
534
          command: .circleci/unittest/windows/scripts/run_test.sh
moto's avatar
moto committed
535
536
      - run:
          name: Post process
537
          command: .circleci/unittest/windows/scripts/post_process.sh
moto's avatar
moto committed
538
539
      - store_test_results:
          path: test-results
moto's avatar
moto committed
540

moto's avatar
moto committed
541
542
543
  unittest_macos_cpu:
    <<: *binary_common
    macos:
544
      xcode: "9.4.1"
moto's avatar
moto committed
545
546
547
548
549
550
    resource_class: large
    steps:
      - checkout
      - install_build_tools_macos
      - attach_workspace:
          at: third_party
551
      - designate_upload_channel
moto's avatar
moto committed
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
      - generate_cache_key
      - restore_cache:
          {% raw %}
          keys:
            - env-v3-macos-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
          {% endraw %}
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:
          {% raw %}
          key: env-v3-macos-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum "third_party/CMakeLists.txt" }}-{{ checksum ".cachekey" }}
          {% endraw %}
          paths:
            - conda
            - env
            - third_party/build
            - third_party/install
            - third_party/src
      - run:
          name: Install torchaudio
          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

583
584
585
586
587
588
589
  stylecheck:
    <<: *binary_common
    docker:
      - image: "pytorch/torchaudio_unittest_base:manylinux"
    resource_class: medium
    steps:
      - checkout
590
      - designate_upload_channel
591
      - generate_cache_key
592
593
594
      - restore_cache:
          {% raw %}
          keys:
595
            - env-v2-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
596
597
598
599
600
601
          {% endraw %}
      - run:
          name: Setup
          command: .circleci/unittest/linux/scripts/setup_env.sh
      - save_cache:
          {% raw %}
602
          key: env-v2-linux-{{ arch }}-py<< parameters.python_version >>-{{ checksum ".circleci/unittest/linux/scripts/environment.yml" }}-{{ checksum ".cachekey" }}
603
604
605
606
607
          {% endraw %}
          paths:
            - conda
            - env
            - third_party/build
moto's avatar
moto committed
608
609
            - third_party/install
            - third_party/src
610
611
612
613
      - run:
          name: Run style check
          command: .circleci/unittest/linux/scripts/run_style_checks.sh

614
615
616
  torchscript_bc_test:
    docker:
      - image: "pytorch/torchaudio_unittest_base:manylinux"
617
    resource_class: large
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
    steps:
      - checkout
      - generate_cache_key
      - restore_cache:
          {% raw %}
          keys:
            - torchscript-bc-test-release-envs-v1-{{ arch }}-{{ checksum ".cachekey" }}
          {% endraw %}
      - run:
          name: Generate Objects
          command: |
             .circleci/torchscript_bc_test/setup_release_envs.sh
             .circleci/torchscript_bc_test/generate_objects.sh
      - save_cache:
          {% raw %}
          key: torchscript-bc-test-release-envs-v1-{{ arch }}-{{ checksum ".cachekey" }}
          {% endraw %}
          paths:
            - conda
            - envs
      - store_artifacts:
          path: test/torchscript_bc_test/assets
      - persist_to_workspace:
          root: .
          paths:
            - test/torchscript_bc_test/assets
      - run:
          name: Run BC check test
          command: |
            .circleci/torchscript_bc_test/setup_master_envs.sh
            .circleci/torchscript_bc_test/validate_objects.sh

Edward Z. Yang's avatar
Edward Z. Yang committed
650
651
652
653
workflows:
  build:
    jobs:
      - circleci_consistency
moto's avatar
moto committed
654
655
656
      {{ build_workflows() }}
  unittest:
    jobs:
657
658
659
      - torchscript_bc_test:
          requires:
          - download_third_parties_nix
moto's avatar
moto committed
660
      {{ unittest_workflows() }}
Edward Z. Yang's avatar
Edward Z. Yang committed
661
662
  nightly:
    jobs:
moto's avatar
moto committed
663
664
665
666
      - circleci_consistency:
          filters:
            branches:
              only: nightly
moto's avatar
moto committed
667
      {{ build_workflows(prefix="nightly_", filter_branch="nightly", upload=True) }}
Mingbo Wan's avatar
Mingbo Wan committed
668
669
670
671
672
673
674
675
676
  docker_build:
    triggers:
      - schedule:
          cron: "0 10 * * 0"
          filters:
            branches:
              only:
                - master
    jobs:
677
      - smoke_test_docker_image_build