|
1 | 1 | [dev] |
2 | | -# Set to "huggingface", for example, if you are a huggingface developer. Default is "" |
3 | 2 | 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) |
6 | 3 | 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) |
9 | 4 | neuron_mode = false |
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 | 5 | 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) |
15 | 6 | 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) |
18 | 7 | 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) |
21 | 8 | 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 |
25 | 9 | 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 |
29 | 10 | 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. |
35 | 11 | deep_canary_mode = false |
36 | 12 |
|
37 | 13 | [build] |
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. |
43 | | -build_training = true |
| 14 | +build_frameworks = [ "huggingface_pytorch",] |
| 15 | +build_training = false |
44 | 16 | 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" |
48 | 17 | do_build = true |
49 | 18 |
|
50 | 19 | [notify] |
51 | | -### Notify on test failures |
52 | | -### Off by default |
53 | 20 | notify_test_failures = false |
54 | | - # Valid values: medium or high |
55 | | - notification_severity = "medium" |
| 21 | +notification_severity = "medium" |
56 | 22 |
|
57 | 23 | [test] |
58 | | -### On by default |
59 | 24 | sanity_tests = true |
60 | 25 | security_tests = true |
61 | | - safety_check_test = false |
62 | | - ecr_scan_allowlist_feature = false |
| 26 | +safety_check_test = false |
| 27 | +ecr_scan_allowlist_feature = false |
63 | 28 | ecs_tests = true |
64 | 29 | eks_tests = true |
65 | 30 | ec2_tests = true |
66 | | -# Set it to true if you are preparing a Benchmark related PR |
67 | 31 | 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) |
73 | 32 | ec2_tests_on_heavy_instances = false |
74 | | - |
75 | | -### SM specific tests |
76 | | -### On by default |
77 | 33 | sagemaker_local_tests = true |
78 | | - |
79 | | -# run standard sagemaker remote tests from test/sagemaker_tests |
80 | 34 | sagemaker_remote_tests = true |
81 | | -# run efa sagemaker tests |
82 | 35 | sagemaker_efa_tests = false |
83 | | -# run release_candidate_integration tests |
84 | 36 | sagemaker_rc_tests = false |
85 | | -# run sagemaker benchmark tests |
86 | 37 | sagemaker_benchmark_tests = false |
87 | | - |
88 | | -# SM remote EFA test instance type |
89 | 38 | sagemaker_remote_efa_instance_type = "" |
90 | | - |
91 | | -# Run CI tests for nightly images |
92 | | -# false by default |
93 | 39 | nightly_pr_test_mode = false |
94 | | - |
95 | 40 | use_scheduler = false |
96 | 41 |
|
97 | 42 | [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 |
107 | 43 | dlc-pr-mxnet-training = "" |
108 | 44 | dlc-pr-pytorch-training = "" |
109 | 45 | dlc-pr-tensorflow-2-training = "" |
110 | 46 | dlc-pr-autogluon-training = "" |
111 | | - |
112 | | -# HuggingFace Training |
113 | 47 | dlc-pr-huggingface-tensorflow-training = "" |
114 | 48 | dlc-pr-huggingface-pytorch-training = "" |
115 | | - |
116 | | -# Training Compiler |
117 | 49 | dlc-pr-huggingface-pytorch-trcomp-training = "" |
118 | 50 | dlc-pr-huggingface-tensorflow-2-trcomp-training = "" |
119 | 51 | dlc-pr-pytorch-trcomp-training = "" |
120 | | - |
121 | | -# Neuron Training |
122 | 52 | dlc-pr-mxnet-neuron-training = "" |
123 | 53 | dlc-pr-pytorch-neuron-training = "" |
124 | 54 | dlc-pr-tensorflow-2-neuron-training = "" |
125 | | - |
126 | | -# Stability AI Training |
127 | 55 | dlc-pr-stabilityai-pytorch-training = "" |
128 | | - |
129 | | -# Habana Training |
130 | 56 | dlc-pr-pytorch-habana-training = "" |
131 | 57 | dlc-pr-tensorflow-2-habana-training = "" |
132 | | - |
133 | | -### INFERENCE PR JOBS ### |
134 | | - |
135 | | -# Standard Framework Inference |
136 | 58 | dlc-pr-mxnet-inference = "" |
137 | 59 | dlc-pr-pytorch-inference = "" |
138 | 60 | dlc-pr-tensorflow-2-inference = "" |
139 | 61 | dlc-pr-autogluon-inference = "" |
140 | | - |
141 | | -# Neuron Inference |
142 | 62 | dlc-pr-mxnet-neuron-inference = "" |
143 | 63 | dlc-pr-pytorch-neuron-inference = "" |
144 | 64 | dlc-pr-tensorflow-1-neuron-inference = "" |
145 | 65 | dlc-pr-tensorflow-2-neuron-inference = "" |
146 | | - |
147 | | -# HuggingFace Inference |
148 | 66 | dlc-pr-huggingface-tensorflow-inference = "" |
149 | | -dlc-pr-huggingface-pytorch-inference = "" |
| 67 | +dlc-pr-huggingface-pytorch-inference = "huggingface/pytorch/inference/buildspec-2-1-0.yml" |
150 | 68 | dlc-pr-huggingface-pytorch-neuron-inference = "" |
151 | | - |
152 | | -# Stability AI Inference |
153 | 69 | dlc-pr-stabilityai-pytorch-inference = "" |
154 | | - |
155 | | -# Graviton Inference |
156 | 70 | dlc-pr-mxnet-graviton-inference = "" |
157 | 71 | dlc-pr-pytorch-graviton-inference = "" |
158 | 72 | dlc-pr-tensorflow-2-graviton-inference = "" |
159 | | - |
160 | | -# ARM64 Inference |
161 | 73 | dlc-pr-pytorch-arm64-inference = "" |
162 | 74 | dlc-pr-tensorflow-2-arm64-inference = "" |
163 | | - |
164 | | -# EIA Inference |
165 | 75 | dlc-pr-mxnet-eia-inference = "" |
166 | 76 | dlc-pr-pytorch-eia-inference = "" |
167 | 77 | dlc-pr-tensorflow-2-eia-inference = "" |
| 78 | + |
0 commit comments