.ci/aws: Add g4dn testing to PR CI

Signed-off-by: Seth Zegelstein <szegel@amazon.com>
a-szegel · Aug 12, 2024 · ff0ecc5 · ff0ecc5
1 parent 59bfaa4
commit ff0ecc5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.ci/aws/Jenkinsfile b/.ci/aws/Jenkinsfile
@@ -230,6 +230,10 @@ pipeline {
                     def trn1n_region = "us-east-1"
                     def trn1n_odcr = "cr-07342cf6439332dce"
                     def trn1n_addl_args = "${base_args} --odcr-placement-group-name trn1n-placement-group --test-list test_nccom_test"
+                    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)
@@ -252,6 +256,9 @@ pipeline {
                     // trn1n tests
                     stages["4_trn1n_ubuntu2004"] = get_test_stage_with_lock("4_trn1n_ubuntu2004", env.BUILD_TAG, "ubuntu2004", "trn1n.32xlarge", trn1n_region, trn1n_lock_label, num_instances, trn1n_odcr, trn1n_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
                 }
             }