parameterize machine_mem_gb for ancestral_traits task (#427)

broadinstitute · Jun 10, 2022 · 34c865e · 34c865e
1 parent 2bfde65
commit 34c865e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pipes/WDL/tasks/tasks_nextstrain.wdl b/pipes/WDL/tasks/tasks_nextstrain.wdl
@@ -1347,6 +1347,7 @@ task ancestral_traits {
         File?         weights
         Float?        sampling_bias_correction
 
+        Int?          machine_mem_gb
         String        docker = "nextstrain/base:build-20211012T204409Z"
     }
     String out_basename = basename(tree, '.nwk')
@@ -1367,7 +1368,7 @@ task ancestral_traits {
     >>>
     runtime {
         docker: docker
-        memory: "32 GB"
+        memory: select_first([machine_mem_gb, 32]) + " GB"
         cpu :   4
         disks:  "local-disk 100 HDD"
         dx_instance_type: "mem3_ssd1_v2_x4"