Skip to content

Commit 8c8f8ac

Browse files
test pytorch 2.7 EC2 arm64
1 parent 39970a9 commit 8c8f8ac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dlc_developer_config.toml

Lines changed: 2 additions & 2 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 = false
18+
arm64_mode = true
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
@@ -122,7 +122,7 @@ use_scheduler = false
122122
dlc-pr-base = ""
123123

124124
# Standard Framework Training
125-
dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-sm.yml"
125+
dlc-pr-pytorch-training = "pytorch/training/buildspec-arm64-2-7-ec2.yml"
126126
dlc-pr-tensorflow-2-training = ""
127127
dlc-pr-autogluon-training = ""
128128

0 commit comments

Comments
 (0)