diff options
author | Martin Polden <mpolden@mpolden.no> | 2022-11-18 10:36:56 +0100 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2022-11-18 11:03:51 +0100 |
commit | 156efc236b11101f71c61dde943910e510c1ebbe (patch) | |
tree | c834f604c5ba268d4a409d5dc8a82c573f9c1d15 /node-repository/src/test/java/com | |
parent | de69ac37b4df3f1dfce7f442ca9510e778fa1f05 (diff) |
Serialize GPU resources
Diffstat (limited to 'node-repository/src/test/java/com')
-rw-r--r-- | node-repository/src/test/java/com/yahoo/vespa/hosted/provision/persistence/NodeSerializerTest.java | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/persistence/NodeSerializerTest.java b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/persistence/NodeSerializerTest.java index 2536d6b97ca..e8155234989 100644 --- a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/persistence/NodeSerializerTest.java +++ b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/persistence/NodeSerializerTest.java @@ -59,7 +59,7 @@ import static org.junit.Assert.assertTrue; */ public class NodeSerializerTest { - private final NodeFlavors nodeFlavors = FlavorConfigBuilder.createDummies("default", "large", "ugccloud-container", "arm64"); + private final NodeFlavors nodeFlavors = FlavorConfigBuilder.createDummies("default", "large", "ugccloud-container", "arm64", "gpu"); private final NodeSerializer nodeSerializer = new NodeSerializer(nodeFlavors, 1000); private final ManualClock clock = new ManualClock(); @@ -79,7 +79,8 @@ public class NodeSerializerTest { public void reserved_node_serialization() { Node node = createNode(); NodeResources requestedResources = new NodeResources(1.2, 3.4, 5.6, 7.8, - DiskSpeed.any, StorageType.any, Architecture.arm64); + DiskSpeed.any, StorageType.any, Architecture.arm64, + new NodeResources.GpuResources(1, 16)); assertEquals(0, node.history().events().size()); node = node.allocate(ApplicationId.from(TenantName.from("myTenant"), |