summaryrefslogtreecommitdiffstats
path: root/config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2022-11-17 22:37:48 +0100
committerGitHub <noreply@github.com>2022-11-17 22:37:48 +0100
commit2872e68c64605c04147fbf930d76f09078ce5550 (patch)
tree23e074299e8f28bbd4d2c339acf9d289fe621576 /config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java
parentde2362a8a5559c4944a198712796bf458d758b97 (diff)
parentde749a8872fd1d96b7ae7b194b80764abe2768ed (diff)
Merge pull request #24909 from vespa-engine/mpolden/gpu-resources
Support GPU in node specification
Diffstat (limited to 'config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java')
-rw-r--r--config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java3
1 files changed, 2 insertions, 1 deletions
diff --git a/config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java b/config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java
index 313ae0d99ed..c9b945ec79c 100644
--- a/config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java
+++ b/config-provisioning/src/main/java/com/yahoo/config/provision/Flavor.java
@@ -38,7 +38,8 @@ public class Flavor {
flavorConfig.bandwidth() / 1000,
flavorConfig.fastDisk() ? NodeResources.DiskSpeed.fast : NodeResources.DiskSpeed.slow,
flavorConfig.remoteStorage() ? NodeResources.StorageType.remote : NodeResources.StorageType.local,
- NodeResources.Architecture.valueOf(flavorConfig.architecture())),
+ NodeResources.Architecture.valueOf(flavorConfig.architecture()),
+ new NodeResources.GpuResources(flavorConfig.gpuCount(), flavorConfig.gpuMemoryGb())),
Optional.empty(),
Type.valueOf(flavorConfig.environment()),
true,