Skip to content

Commit 255ffab

Browse files
test sm local
1 parent bbbed99 commit 255ffab

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

dlc_developer_config.toml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ neuronx_mode = false
1515
graviton_mode = false
1616
# Please only set it to true if you are preparing a ARM64 related PR
1717
# Do remember to revert it back to false before merging any PR (including ARM64 dedicated PR)
18-
arm64_mode = true
18+
arm64_mode = false
1919
# Please only set it to True if you are preparing a HABANA related PR
2020
# Do remember to revert it back to False before merging any PR (including HABANA dedicated PR)
2121
habana_mode = false
@@ -37,12 +37,12 @@ deep_canary_mode = false
3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
3939
# available frameworks - ["base", "vllm", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = ["pytorch"]
40+
build_frameworks = ["tensorflow"]
4141

4242

4343
# By default we build both training and inference containers. Set true/false values to determine which to build.
44-
build_training = false
45-
build_inference = true
44+
build_training = true
45+
build_inference = false
4646

4747
# Set do_build to "false" to skip builds and test the latest image built by this PR
4848
# Note: at least one build is required to set do_build to "false"
@@ -123,7 +123,7 @@ dlc-pr-base = ""
123123

124124
# Standard Framework Training
125125
dlc-pr-pytorch-training = ""
126-
dlc-pr-tensorflow-2-training = ""
126+
dlc-pr-tensorflow-2-training = "tensorflow/training/buildspec-2-19-sm.yml"
127127
dlc-pr-autogluon-training = ""
128128

129129
# ARM64 Training
@@ -152,7 +152,7 @@ dlc-pr-tensorflow-2-habana-training = ""
152152
### INFERENCE PR JOBS ###
153153

154154
# Standard Framework Inference
155-
dlc-pr-pytorch-inference = "pytorch/inference/buildspec-arm64-2-6-sm.yml"
155+
dlc-pr-pytorch-inference = ""
156156
dlc-pr-tensorflow-2-inference = ""
157157
dlc-pr-autogluon-inference = ""
158158

0 commit comments

Comments
 (0)