diff options
Diffstat (limited to 'config-provisioning')
| -rw-r--r-- | config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java b/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java index 0d21b155571..c1eb3be4275 100644 --- a/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java +++ b/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java @@ -157,6 +157,10 @@ public class NodeResources { return new NodeResources.GpuResources(1, thisMem - otherMem); } + public GpuResources multipliedBy(double factor) { + return new GpuResources(this.count, this.memoryGb * factor); + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -329,10 +333,12 @@ public class NodeResources { } public NodeResources multipliedBy(double factor) { + if (isUnspecified()) return this; return this.withVcpu(vcpu * factor) .withMemoryGb(memoryGb * factor) .withDiskGb(diskGb * factor) - .withBandwidthGbps(bandwidthGbps * factor); + .withBandwidthGbps(bandwidthGbps * factor) + .with(gpuResources.multipliedBy(factor)); } private boolean isInterchangeableWith(NodeResources other) { |