Read GPU fields when creating node through API

1 files changed, 7 insertions, 0 deletions

diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
index 0d21caa3a06..0f944947440 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
@@ -3,6 +3,7 @@ package com.yahoo.vespa.hosted.provision.restapi;
 
 import com.yahoo.config.provision.NodeResources;
 import com.yahoo.slime.Cursor;
+import com.yahoo.slime.Inspector;
 
 /**
  * @author bratseth
@@ -74,4 +75,10 @@ public class NodeResourcesSerializer {
         };
     }
 
+    public static NodeResources.GpuResources gpuResourcesFromSlime(Inspector gpuObject) {
+        if (!gpuObject.valid()) return NodeResources.GpuResources.getDefault();
+        return new NodeResources.GpuResources((int) gpuObject.field("gpuCount").asLong(),
+                                              gpuObject.field("gpuMemory").asDouble());
+    }
+
 }