From ff0ecc5101558f5eeccb6a0e386a2fac18631698 Mon Sep 17 00:00:00 2001 From: Seth Zegelstein Date: Mon, 12 Aug 2024 16:18:48 +0000 Subject: [PATCH] .ci/aws: Add g4dn testing to PR CI Signed-off-by: Seth Zegelstein --- .ci/aws/Jenkinsfile | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/.ci/aws/Jenkinsfile b/.ci/aws/Jenkinsfile index 41f87b296..c7db5bd0b 100644 --- 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 } }