From 0a48243731b489b7979c4388a9bec444f36f7359 Mon Sep 17 00:00:00 2001 From: Chris Tomkins-Tinch Date: Fri, 2 Oct 2020 09:46:51 -0400 Subject: [PATCH] parameterize beast GPU settings (#158) * parameterize beast GPU settings --- pipes/WDL/tasks/tasks_interhost.wdl | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/pipes/WDL/tasks/tasks_interhost.wdl b/pipes/WDL/tasks/tasks_interhost.wdl index 675d74b31..9a50fa1dc 100644 --- a/pipes/WDL/tasks/tasks_interhost.wdl +++ b/pipes/WDL/tasks/tasks_interhost.wdl @@ -89,6 +89,11 @@ task multi_align_mafft { task beast { input { File beauti_xml + + String? accelerator_type + Int? accelerator_count + String? gpu_type + Int? gpu_count String docker="quay.io/broadinstitute/beast-beagle-cuda:1.10.5pre" } @@ -123,10 +128,10 @@ task beast { gpu: true # dxWDL dx_timeout: "40H" # dxWDL dx_instance_type: "mem1_ssd1_gpu2_x8" # dxWDL - acceleratorType: "nvidia-tesla-k80" # GCP PAPIv2 - acceleratorCount: 4 # GCP PAPIv2 - gpuType: "nvidia-tesla-k80" # Terra - gpuCount: 4 # Terra + acceleratorType: select_first([accelerator_type, "nvidia-tesla-k80"]) # GCP PAPIv2 + acceleratorCount: select_first([accelerator_count, 4]) # GCP PAPIv2 + gpuType: select_first([gpu_type, "nvidia-tesla-k80"]) # Terra + gpuCount: select_first([gpu_count, 4]) # Terra nvidiaDriverVersion: "410.79" } }