Skip to content

Commit

Permalink
testsys: Add support for gpu based ecs workloads
Browse files Browse the repository at this point in the history
  • Loading branch information
ecpullen committed Aug 17, 2023
1 parent 947c178 commit 101d813
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tools/testsys/src/aws_ecs.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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() {
Expand Down

0 comments on commit 101d813

Please sign in to comment.