From 12d9f02f05555b27a4f936b28c11f618eddabae3 Mon Sep 17 00:00:00 2001 From: Ola Aunronning Date: Wed, 20 Sep 2023 10:29:33 +0200 Subject: Fix GpuResources::plus --- .../src/main/java/com/yahoo/config/provision/NodeResources.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (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 a431dd61b0d..ae5b64ee013 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,7 +144,9 @@ public class NodeResources { public static GpuResources getDefault() { return zero; } public GpuResources plus(GpuResources other) { - return new GpuResources(this.count + other.count, this.memoryGb + other.memoryGb); + var thisMem = this.count() * this.memoryGb(); + var otherMem = other.count() * other.memoryGb(); + return new NodeResources.GpuResources(1, thisMem + otherMem); } public GpuResources minus(GpuResources other) { -- cgit v1.2.3