add rusage

weng-lab · Oct 21, 2023 · 21ccf21 · 21ccf21
1 parent b482bbb
commit 21ccf21
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -13,7 +13,7 @@ plugins {
 }
 
 group = "io.krews"
-version = "0.14.7"
+version = "0.14.8"
 
 repositories {
     maven { setUrl("https://dl.bintray.com/kotlin/kotlin-eap") }

diff --git a/src/main/kotlin/krews/config/BsubConfig.kt b/src/main/kotlin/krews/config/BsubConfig.kt
@@ -22,5 +22,7 @@ data class BsubTaskConfig(
     // SBatch partition to use.
     val partition: String? = null,
     // Additional bsub arguments
-    val sbatchArgs: Map<String, String>? = null
+    val sbatchArgs: Map<String, String>? = null,
+    // Additional rusage parameters
+    val rUsage: String? = null
 )
diff --git a/src/main/kotlin/krews/executor/bsub/BsubExecutor.kt b/src/main/kotlin/krews/executor/bsub/BsubExecutor.kt
@@ -81,7 +81,7 @@ class BsubExecutor(private val workflowConfig: WorkflowConfig) : LocallyDirected
         appendBsubParam(bsubScript, "J", bsubWorkflowJobName)
         appendBsubParam(bsubScript, "o", logsPath.resolve("out.txt"))
         appendBsubParam(bsubScript, "e", logsPath.resolve("err.txt"))
-        appendBsubParam(bsubScript, "M", mem)
+        appendBsubParam(bsubScript, "R", taskConfig.bsub?.rUsage)
         appendBsubParam(bsubScript, "n", cpus)
         appendBsubParam(bsubScript, "W", time)
         appendBsubParam(bsubScript, "q", taskConfig.bsub?.partition)