11[dev ]
2+ # Set to "huggingface", for example, if you are a huggingface developer. Default is ""
23partner_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)
36ei_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)
49neuron_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)
512neuronx_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)
615graviton_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)
718arm64_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)
821habana_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
925huggingface_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
1029trcomp_mode = false
1130# Set deep_canary_mode to true to simulate Deep Canary Test conditions on PR for all frameworks in the
1231# build_frameworks list below. This will cause all image builds and non-deep-canary tests on the PR to be skipped,
@@ -16,68 +35,133 @@ trcomp_mode = false
1635deep_canary_mode = false
1736
1837[build ]
19- build_frameworks = [ " pytorch" ,]
20- build_training = false
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
2144build_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"
2248do_build = true
2349
2450[notify ]
51+ # ## Notify on test failures
52+ # ## Off by default
2553notify_test_failures = false
26- notification_severity = " medium"
54+ # Valid values: medium or high
55+ notification_severity = " medium"
2756
2857[test ]
29- sanity_tests = false
30- security_tests = false
31- safety_check_test = false
32- ecr_scan_allowlist_feature = false
33- ecs_tests = false
34- eks_tests = false
58+ # ## On by default
59+ sanity_tests = true
60+ security_tests = true
61+ safety_check_test = false
62+ ecr_scan_allowlist_feature = false
63+ ecs_tests = true
64+ eks_tests = true
3565ec2_tests = true
66+ # Set it to true if you are preparing a Benchmark related PR
3667ec2_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)
3773ec2_tests_on_heavy_instances = false
38- sagemaker_local_tests = false
39- sagemaker_remote_tests = false
74+
75+ # ## SM specific tests
76+ # ## On by default
77+ sagemaker_local_tests = true
78+
79+ # run standard sagemaker remote tests from test/sagemaker_tests
80+ sagemaker_remote_tests = true
81+ # run efa sagemaker tests
4082sagemaker_efa_tests = false
83+ # run release_candidate_integration tests
4184sagemaker_rc_tests = false
85+ # run sagemaker benchmark tests
4286sagemaker_benchmark_tests = false
87+
88+ # SM remote EFA test instance type
4389sagemaker_remote_efa_instance_type = " "
90+
91+ # Run CI tests for nightly images
92+ # false by default
4493nightly_pr_test_mode = false
94+
4595use_scheduler = false
4696
4797[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
48107dlc-pr-mxnet-training = " "
49108dlc-pr-pytorch-training = " "
50109dlc-pr-tensorflow-2-training = " "
51110dlc-pr-autogluon-training = " "
111+
112+ # HuggingFace Training
52113dlc-pr-huggingface-tensorflow-training = " "
53114dlc-pr-huggingface-pytorch-training = " "
115+
116+ # Training Compiler
54117dlc-pr-huggingface-pytorch-trcomp-training = " "
55118dlc-pr-huggingface-tensorflow-2-trcomp-training = " "
56119dlc-pr-pytorch-trcomp-training = " "
120+
121+ # Neuron Training
57122dlc-pr-mxnet-neuron-training = " "
58123dlc-pr-pytorch-neuron-training = " "
59124dlc-pr-tensorflow-2-neuron-training = " "
125+
126+ # Stability AI Training
60127dlc-pr-stabilityai-pytorch-training = " "
128+
129+ # Habana Training
61130dlc-pr-pytorch-habana-training = " "
62131dlc-pr-tensorflow-2-habana-training = " "
132+
133+ # ## INFERENCE PR JOBS ###
134+
135+ # Standard Framework Inference
63136dlc-pr-mxnet-inference = " "
64- dlc-pr-pytorch-inference = " pytorch/inference/buildspec-2-4-ec2.yml "
137+ dlc-pr-pytorch-inference = " "
65138dlc-pr-tensorflow-2-inference = " "
66139dlc-pr-autogluon-inference = " "
140+
141+ # Neuron Inference
67142dlc-pr-mxnet-neuron-inference = " "
68143dlc-pr-pytorch-neuron-inference = " "
69144dlc-pr-tensorflow-1-neuron-inference = " "
70145dlc-pr-tensorflow-2-neuron-inference = " "
146+
147+ # HuggingFace Inference
71148dlc-pr-huggingface-tensorflow-inference = " "
72149dlc-pr-huggingface-pytorch-inference = " "
73150dlc-pr-huggingface-pytorch-neuron-inference = " "
151+
152+ # Stability AI Inference
74153dlc-pr-stabilityai-pytorch-inference = " "
154+
155+ # Graviton Inference
75156dlc-pr-mxnet-graviton-inference = " "
76157dlc-pr-pytorch-graviton-inference = " "
77158dlc-pr-tensorflow-2-graviton-inference = " "
159+
160+ # ARM64 Inference
78161dlc-pr-pytorch-arm64-inference = " "
79162dlc-pr-tensorflow-2-arm64-inference = " "
163+
164+ # EIA Inference
80165dlc-pr-mxnet-eia-inference = " "
81166dlc-pr-pytorch-eia-inference = " "
82167dlc-pr-tensorflow-2-eia-inference = " "
83-
0 commit comments