.ci/aws: Add p3dn testing to Jenkins

Adds ubuntu2004, and ubuntu2204 p3dn testing. Signed-off-by: Seth Zegelstein <[email protected]>
a-szegel · May 14, 2024 · 0dae49e · 0dae49e
1 parent 3d88b88
commit 0dae49e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.ci/aws/Jenkinsfile b/.ci/aws/Jenkinsfile
@@ -58,6 +58,10 @@ pipeline {
                     def addl_args_pr = "--test-aws-ofi-nccl-pr $env.CHANGE_ID --test-nccl-version ${nccl_version}"
                     def config = ".ci/aws/aws_ofi_nccl_pr_ci.yaml"
                     def num_instances = 4
+                    def p3dn_lock_label = "p3dn-1-4node"
+                    def p3dn_region = "us-east-1"
+                    def p3dn_odcr = "cr-0ca89ca67f047efa8"
+                    def p3dn_addl_args = "${addl_args_pr} --odcr ${p3dn_odcr} --odcr-placement-group-name efa-placement-group2"
                     def p4d_lock_label = "p4d-1-4node"
                     def p4d_region = "us-east-2"
                     def p4d_odcr = "cr-0e5eebb3c896f6af0"
@@ -67,6 +71,10 @@ pipeline {
                     def p5_odcr = "cr-0b5693c31bc89b82d"
                     def p5_addl_args = "${addl_args_pr} --odcr ${p5_odcr}"
 
+                    // p3dn tests
+                    stages["4_p3dn_ubuntu2004"] = common.get_test_stage_with_lock("4_p3dn_ubuntu2004", env.BUILD_TAG, "ubuntu2004", "p3dn.24xlarge", p3dn_region, p3dn_lock_label, num_instances, config, p3dn_addl_args)
+                    stages["4_p3dn_ubuntu2204"] = common.get_test_stage_with_lock("4_p3dn_ubuntu2204", env.BUILD_TAG, "ubuntu2204", "p3dn.24xlarge", p3dn_region, p3dn_lock_label, num_instances, config, p3dn_addl_args)
+
                     // p4d tests
                     stages["4_p4d_alinux2"] = common.get_test_stage_with_lock("4_p4d_alinux2", env.BUILD_TAG, "alinux2", "p4d.24xlarge", p4d_region, p4d_lock_label, num_instances, config, p4d_addl_args)
                     stages["4_p4d_ubuntu2004"] = common.get_test_stage_with_lock("4_p4d_ubuntu2004", env.BUILD_TAG, "ubuntu2004", "p4d.24xlarge", p4d_region, p4d_lock_label, num_instances, config, p4d_addl_args)