#328: TonY shouldn't throw GPUDiscoverer errors when GPUs are not req…

…uested (#329)
tony-framework · Jun 17, 2019 · 1d8f077 · 1d8f077
1 parent 42db0c6
commit 1d8f077
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/build.gradle b/build.gradle
@@ -80,7 +80,7 @@ ext.deps = [
 
 allprojects {
   group = "com.linkedin.tony"
-  project.version = "0.3.15"
+  project.version = "0.3.16"
 }
 
 task sourcesJar(type: Jar) {

diff --git a/tony-core/src/main/java/com/linkedin/tony/TaskMonitor.java b/tony-core/src/main/java/com/linkedin/tony/TaskMonitor.java
@@ -81,8 +81,8 @@ void initMetrics() {
 
   private boolean checkIsGpuMachine(Configuration conf) {
     int numWorkerGpus = conf.getInt(
-        TonyConfigurationKeys.getResourceKey("worker", "gpus"), 0);
-    LOG.info("Number of GPUs requested: " + numWorkerGpus);
+        TonyConfigurationKeys.getResourceKey(taskType, "gpus"), 0);
+    LOG.info("Number of GPUs requested for " + taskType + ": " + numWorkerGpus);
     return numWorkerGpus > 0;
   }