diff options
author | Bjørn Christian Seime <bjorn.christian@seime.no> | 2023-09-25 14:53:50 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-25 14:53:50 +0200 |
commit | 27472f94770a6644d44b765cedf802d8bb38ac03 (patch) | |
tree | d219d2dc12f34f5e93ff8b3c03c1fc6ac8abf238 /config-model-api/src/main | |
parent | 7facdd6177063f772c497000b9c12e4653a2db83 (diff) | |
parent | 2a537e9ce9223110ca2bbedd7e88139c24524049 (diff) |
Merge pull request #28645 from vespa-engine/bjorncs/analyze-model
Bjorncs/analyze model
Diffstat (limited to 'config-model-api/src/main')
-rw-r--r-- | config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java b/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java index 422ceba8074..595cd97e6b6 100644 --- a/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java +++ b/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java @@ -4,6 +4,7 @@ package com.yahoo.config.model.api; import com.yahoo.config.ModelReference; import com.yahoo.config.application.api.ApplicationFile; +import com.yahoo.config.application.api.ApplicationPackage; import com.yahoo.config.application.api.DeployLogger; /** @@ -11,7 +12,7 @@ import com.yahoo.config.application.api.DeployLogger; */ public interface OnnxModelCost { - Calculator newCalculator(DeployLogger logger); + Calculator newCalculator(ApplicationPackage appPkg, DeployLogger logger); interface Calculator { long aggregatedModelCostInBytes(); @@ -20,7 +21,7 @@ public interface OnnxModelCost { } static OnnxModelCost disabled() { - return (__) -> new Calculator() { + return (__, ___) -> new Calculator() { @Override public long aggregatedModelCostInBytes() { return 0; } @Override public void registerModel(ApplicationFile path) {} @Override public void registerModel(ModelReference ref) {} |