|
1 | 1 | [dev] |
| 2 | +# Set to "huggingface", for example, if you are a huggingface developer. Default is "" |
2 | 3 | partner_developer = "" |
| 4 | +# Please only set it to true if you are preparing an EI related PR |
| 5 | +# Do remember to revert it back to false before merging any PR (including EI dedicated PR) |
3 | 6 | ei_mode = false |
| 7 | +# Please only set it to true if you are preparing a NEURON related PR |
| 8 | +# Do remember to revert it back to false before merging any PR (including NEURON dedicated PR) |
4 | 9 | neuron_mode = false |
5 | | -neuronx_mode = true |
| 10 | +# Please only set it to true if you are preparing a NEURONX related PR |
| 11 | +# Do remember to revert it back to false before merging any PR (including NEURONX dedicated PR) |
| 12 | +neuronx_mode = false |
| 13 | +# Please only set it to true if you are preparing a GRAVITON related PR |
| 14 | +# Do remember to revert it back to false before merging any PR (including GRAVITON dedicated PR) |
6 | 15 | graviton_mode = false |
| 16 | +# Please only set it to true if you are preparing a ARM64 related PR |
| 17 | +# Do remember to revert it back to false before merging any PR (including ARM64 dedicated PR) |
7 | 18 | arm64_mode = false |
| 19 | +# Please only set it to True if you are preparing a HABANA related PR |
| 20 | +# Do remember to revert it back to False before merging any PR (including HABANA dedicated PR) |
8 | 21 | habana_mode = false |
| 22 | +# Please only set it to True if you are preparing a HUGGINGFACE TRCOMP related PR |
| 23 | +# Do remember to revert it back to False before merging any PR (including HUGGINGFACE TRCOMP dedicated PR) |
| 24 | +# This mode is used to build TF 2.6 and PT1.11 DLC |
9 | 25 | huggingface_trcomp_mode = false |
| 26 | +# Please only set it to True if you are preparing a TRCOMP related PR |
| 27 | +# Do remember to revert it back to False before merging any PR (including TRCOMP dedicated PR) |
| 28 | +# This mode is used to build PT1.12 and above DLC |
10 | 29 | trcomp_mode = false |
| 30 | +# Set deep_canary_mode to true to simulate Deep Canary Test conditions on PR for all frameworks in the |
| 31 | +# build_frameworks list below. This will cause all image builds and non-deep-canary tests on the PR to be skipped, |
| 32 | +# regardless of whether they are enabled or disabled below. |
| 33 | +# Set graviton_mode/arm64_mode to true to run Deep Canaries on Graviton/ARM64 images. |
| 34 | +# Do remember to revert it back to false before merging any PR. |
11 | 35 | deep_canary_mode = false |
12 | 36 |
|
13 | 37 | [build] |
14 | | -build_frameworks = [ "huggingface_pytorch",] |
| 38 | +# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. |
| 39 | +# available frameworks - ["autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "mxnet", "pytorch", "stabilityai_pytorch"] |
| 40 | +build_frameworks = [] |
| 41 | + |
| 42 | +# By default we build both training and inference containers. Set true/false values to determine which to build. |
15 | 43 | build_training = true |
16 | | -build_inference = false |
| 44 | +build_inference = true |
| 45 | + |
| 46 | +# Set do_build to "false" to skip builds and test the latest image built by this PR |
| 47 | +# Note: at least one build is required to set do_build to "false" |
17 | 48 | do_build = true |
18 | 49 |
|
19 | 50 | [notify] |
| 51 | +### Notify on test failures |
| 52 | +### Off by default |
20 | 53 | notify_test_failures = false |
21 | | -notification_severity = "medium" |
| 54 | + # Valid values: medium or high |
| 55 | + notification_severity = "medium" |
22 | 56 |
|
23 | 57 | [test] |
| 58 | +### On by default |
24 | 59 | sanity_tests = true |
25 | 60 | security_tests = true |
26 | | -safety_check_test = false |
27 | | -ecr_scan_allowlist_feature = false |
| 61 | + safety_check_test = false |
| 62 | + ecr_scan_allowlist_feature = false |
28 | 63 | ecs_tests = true |
29 | 64 | eks_tests = true |
30 | 65 | ec2_tests = true |
| 66 | +# Set it to true if you are preparing a Benchmark related PR |
31 | 67 | ec2_benchmark_tests = false |
| 68 | + |
| 69 | +### Set ec2_tests_on_heavy_instances = true to be able to run any EC2 tests that use large/expensive instance types by |
| 70 | +### default. If false, these types of tests will be skipped while other tests will run as usual. |
| 71 | +### These tests are run in EC2 test jobs, so ec2_tests must be true if ec2_tests_on_heavy_instances is true. |
| 72 | +### Off by default (set to false) |
32 | 73 | ec2_tests_on_heavy_instances = false |
| 74 | + |
| 75 | +### SM specific tests |
| 76 | +### On by default |
33 | 77 | sagemaker_local_tests = true |
| 78 | + |
| 79 | +# run standard sagemaker remote tests from test/sagemaker_tests |
34 | 80 | sagemaker_remote_tests = true |
| 81 | +# run efa sagemaker tests |
35 | 82 | sagemaker_efa_tests = false |
| 83 | +# run release_candidate_integration tests |
36 | 84 | sagemaker_rc_tests = false |
| 85 | +# run sagemaker benchmark tests |
37 | 86 | sagemaker_benchmark_tests = false |
| 87 | + |
| 88 | +# SM remote EFA test instance type |
38 | 89 | sagemaker_remote_efa_instance_type = "" |
| 90 | + |
| 91 | +# Run CI tests for nightly images |
| 92 | +# false by default |
39 | 93 | nightly_pr_test_mode = false |
| 94 | + |
40 | 95 | use_scheduler = false |
41 | 96 |
|
42 | 97 | [buildspec_override] |
| 98 | +# Assign the path to the required buildspec file from the deep-learning-containers folder |
| 99 | +# For example: |
| 100 | +# dlc-pr-tensorflow-2-habana-training = "habana/tensorflow/training/buildspec-2-10.yml" |
| 101 | +# dlc-pr-pytorch-inference = "pytorch/inference/buildspec-1-12.yml" |
| 102 | +# Setting the buildspec file path to "" allows the image builder to choose the default buildspec file. |
| 103 | + |
| 104 | +### TRAINING PR JOBS ### |
| 105 | + |
| 106 | +# Standard Framework Training |
43 | 107 | dlc-pr-mxnet-training = "" |
44 | 108 | dlc-pr-pytorch-training = "" |
45 | 109 | dlc-pr-tensorflow-2-training = "" |
46 | 110 | dlc-pr-autogluon-training = "" |
| 111 | + |
| 112 | +# HuggingFace Training |
47 | 113 | dlc-pr-huggingface-tensorflow-training = "" |
48 | 114 | dlc-pr-huggingface-pytorch-training = "" |
| 115 | + |
| 116 | +# Training Compiler |
49 | 117 | dlc-pr-huggingface-pytorch-trcomp-training = "" |
50 | 118 | dlc-pr-huggingface-tensorflow-2-trcomp-training = "" |
51 | 119 | dlc-pr-pytorch-trcomp-training = "" |
| 120 | + |
| 121 | +# Neuron Training |
52 | 122 | dlc-pr-mxnet-neuron-training = "" |
53 | 123 | dlc-pr-pytorch-neuron-training = "" |
54 | 124 | dlc-pr-tensorflow-2-neuron-training = "" |
| 125 | + |
| 126 | +# Stability AI Training |
55 | 127 | dlc-pr-stabilityai-pytorch-training = "" |
| 128 | + |
| 129 | +# Habana Training |
56 | 130 | dlc-pr-pytorch-habana-training = "" |
57 | 131 | dlc-pr-tensorflow-2-habana-training = "" |
| 132 | + |
| 133 | +### INFERENCE PR JOBS ### |
| 134 | + |
| 135 | +# Standard Framework Inference |
58 | 136 | dlc-pr-mxnet-inference = "" |
59 | 137 | dlc-pr-pytorch-inference = "" |
60 | 138 | dlc-pr-tensorflow-2-inference = "" |
61 | 139 | dlc-pr-autogluon-inference = "" |
| 140 | + |
| 141 | +# Neuron Inference |
62 | 142 | dlc-pr-mxnet-neuron-inference = "" |
63 | 143 | dlc-pr-pytorch-neuron-inference = "" |
64 | 144 | dlc-pr-tensorflow-1-neuron-inference = "" |
65 | 145 | dlc-pr-tensorflow-2-neuron-inference = "" |
| 146 | + |
| 147 | +# HuggingFace Inference |
66 | 148 | dlc-pr-huggingface-tensorflow-inference = "" |
67 | 149 | dlc-pr-huggingface-pytorch-inference = "" |
68 | 150 | dlc-pr-huggingface-pytorch-neuron-inference = "" |
| 151 | + |
| 152 | +# Stability AI Inference |
69 | 153 | dlc-pr-stabilityai-pytorch-inference = "" |
| 154 | + |
| 155 | +# Graviton Inference |
70 | 156 | dlc-pr-mxnet-graviton-inference = "" |
71 | 157 | dlc-pr-pytorch-graviton-inference = "" |
72 | 158 | dlc-pr-tensorflow-2-graviton-inference = "" |
| 159 | + |
| 160 | +# ARM64 Inference |
73 | 161 | dlc-pr-pytorch-arm64-inference = "" |
74 | 162 | dlc-pr-tensorflow-2-arm64-inference = "" |
| 163 | + |
| 164 | +# EIA Inference |
75 | 165 | dlc-pr-mxnet-eia-inference = "" |
76 | 166 | dlc-pr-pytorch-eia-inference = "" |
77 | 167 | dlc-pr-tensorflow-2-eia-inference = "" |
78 | | -# WARNING: Unrecognized key generated below |
79 | | -dlc-pr-huggingface-pytorch-neuronx-training = "huggingface/pytorch/training/buildspec-neuronx.yml" |
80 | | - |
|
0 commit comments