From e25bef15919124ccad19736f9ced8a06e14222a0 Mon Sep 17 00:00:00 2001 From: Ola Aunronning Date: Wed, 20 Sep 2023 11:45:46 +0200 Subject: Consider 'zero' GpuResources when accumulating --- .../src/main/java/com/yahoo/config/provision/NodeResources.java | 2 ++ 1 file changed, 2 insertions(+) (limited to 'config-provisioning/src') 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 f9e6da14493..0d21b155571 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 @@ -144,12 +144,14 @@ public class NodeResources { public static GpuResources getDefault() { return zero; } public GpuResources plus(GpuResources other) { + if (other.isZero()) return this; var thisMem = this.count() * this.memoryGb(); var otherMem = other.count() * other.memoryGb(); return new NodeResources.GpuResources(1, thisMem + otherMem); } public GpuResources minus(GpuResources other) { + if (other.isZero()) return this; var thisMem = this.count() * this.memoryGb(); var otherMem = other.count() * other.memoryGb(); return new NodeResources.GpuResources(1, thisMem - otherMem); -- cgit v1.2.3