diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2021-06-23 10:00:18 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2021-06-23 10:00:18 +0200 |
commit | 8d1451224710fa7a65e9a7410113c64c104a8dd0 (patch) | |
tree | a6f12ad954bfa900f64f5f3c0f4b43a0aa2bcb70 /configserver | |
parent | ed7bc94c5c184b1dc735d8db4f95894102cd46f2 (diff) |
Add feature falg for controlling onnx dryrun verification.
Diffstat (limited to 'configserver')
-rw-r--r-- | configserver/src/main/java/com/yahoo/vespa/config/server/deploy/ModelContextImpl.java | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/configserver/src/main/java/com/yahoo/vespa/config/server/deploy/ModelContextImpl.java b/configserver/src/main/java/com/yahoo/vespa/config/server/deploy/ModelContextImpl.java index 94cfba12453..2b8eaf86cda 100644 --- a/configserver/src/main/java/com/yahoo/vespa/config/server/deploy/ModelContextImpl.java +++ b/configserver/src/main/java/com/yahoo/vespa/config/server/deploy/ModelContextImpl.java @@ -182,6 +182,7 @@ public class ModelContextImpl implements ModelContext { private final int maxMergeQueueSize; private final int largeRankExpressionLimit; private final boolean throwIfResourceLimitsSpecified; + private final boolean dryRunOnnxOnSetup; public FeatureFlags(FlagSource source, ApplicationId appId) { this.dedicatedClusterControllerFlavor = parseDedicatedClusterControllerFlavor(flagValue(source, appId, Flags.DEDICATED_CLUSTER_CONTROLLER_FLAVOR)); @@ -209,6 +210,7 @@ public class ModelContextImpl implements ModelContext { this.maxConcurrentMergesPerContentNode = flagValue(source, appId, Flags.MAX_CONCURRENT_MERGES_PER_NODE); this.maxMergeQueueSize = flagValue(source, appId, Flags.MAX_MERGE_QUEUE_SIZE); this.throwIfResourceLimitsSpecified = flagValue(source, appId, Flags.THROW_EXCEPTION_IF_RESOURCE_LIMITS_SPECIFIED); + this.dryRunOnnxOnSetup = flagValue(source, appId, Flags.DRY_RUN_ONNX_ON_SETUP); } @Override public Optional<NodeResources> dedicatedClusterControllerFlavor() { return Optional.ofNullable(dedicatedClusterControllerFlavor); } @@ -238,6 +240,7 @@ public class ModelContextImpl implements ModelContext { @Override public int maxConcurrentMergesPerNode() { return maxConcurrentMergesPerContentNode; } @Override public int maxMergeQueueSize() { return maxMergeQueueSize; } @Override public boolean throwIfResourceLimitsSpecified() { return throwIfResourceLimitsSpecified; } + @Override public boolean dryRunOnnxOnSetup() { return dryRunOnnxOnSetup; } private static <V> V flagValue(FlagSource source, ApplicationId appId, UnboundFlag<? extends V, ?, ?> flag) { return flag.bindTo(source) |