.ci/aws: Add g4dn testing to PR CI

Signed-off-by: Seth Zegelstein <[email protected]>
a-szegel · Aug 19, 2024 · 428a93c · 428a93c
1 parent 3bd589f
commit 428a93c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.ci/aws/Jenkinsfile b/.ci/aws/Jenkinsfile
@@ -222,6 +222,10 @@ pipeline {
                     def p5_region = "af-south-1"
                     def p5_odcr = "cr-02eb632dcd8175139"
                     def p4_p5_addl_args = "${base_args} --use-prebuilt-ami-with-efa-installer true"
+                    def g4dn_lock_label = "g4dn-1-4node"
+                    def g4dn_region = "us-west-2"
+                    def g4dn_odcr = "cr-0e2f9cac30bb5ad5f"
+                    def g4dn_addl_args = "${base_args} --odcr-placement-group-name g4dn-placement-group"
 
                     // p3dn tests
                     stages["4_p3dn_al2"] = get_test_stage_with_lock("4_p3dn_al2", env.BUILD_TAG, "alinux2", "p3dn.24xlarge", p3dn_region, p3dn_lock_label, num_instances, p3dn_odcr, p3dn_al2_addl_args)
@@ -238,6 +242,9 @@ pipeline {
                     stages["4_p5_ubuntu2004"] = get_test_stage_with_lock("4_p5_ubuntu2004", env.BUILD_TAG, "ubuntu2004", "p5.48xlarge", p5_region, p5_lock_label, num_instances, p5_odcr, p4_p5_addl_args)
                     stages["4_p5_ubuntu2204"] = get_test_stage_with_lock("4_p5_ubuntu2204", env.BUILD_TAG, "ubuntu2204", "p5.48xlarge", p5_region, p5_lock_label, num_instances, p5_odcr, p4_p5_addl_args)
 
+                    // g4dn tests
+                    stages["4_g4dn_ubuntu2204"] = get_test_stage_with_lock("4_g4dn_ubuntu2204", env.BUILD_TAG, "ubuntu2204", "g4dn.12xlarge", g4dn_region, g4dn_lock_label, num_instances, g4dn_odcr, g4dn_addl_args)
+
                     parallel stages
                 }
             }