diff --git a/tools/testsys/src/aws_ecs.rs b/tools/testsys/src/aws_ecs.rs index c95a063b795..2d76e431f4e 100644 --- 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 "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 .map(|(name, image)| WorkloadTest { name: name.to_string(), image: image.to_string(), - ..Default::default() + gpu, }) .collect(); if plugins.is_empty() {