Merge pull request #3347 from ecpullen/ecs-migration-gpu

testsys: Add support for gpu based ecs workloads
bottlerocket-os · Aug 18, 2023 · 112a97b · 112a97b
2 parents 8c4df72 + 101d813
commit 112a97b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/testsys/src/aws_ecs.rs b/tools/testsys/src/aws_ecs.rs
@@ -222,6 +222,7 @@ pub(crate) fn workload_crd(region: &str, test_input: TestInput) -> Result<Test>
         "testsys/type".to_string() => test_input.test_type.to_string(),
         "testsys/cluster".to_string() => cluster_resource_name.to_string(),
     });
+    let gpu = test_input.crd_input.variant.variant_flavor() == Some("nvidia");
     let plugins: Vec<_> = test_input
         .crd_input
         .config
@@ -230,7 +231,7 @@ pub(crate) fn workload_crd(region: &str, test_input: TestInput) -> Result<Test>
         .map(|(name, image)| WorkloadTest {
             name: name.to_string(),
             image: image.to_string(),
-            ..Default::default()
+            gpu,
         })
         .collect();
     if plugins.is_empty() {