diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-09-25 13:33:12 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-09-25 13:33:12 +0200 |
commit | 2a537e9ce9223110ca2bbedd7e88139c24524049 (patch) | |
tree | 4d7a50e55b08b49582a1eabf34524aaa32803450 /config-model-api/src | |
parent | 1dc99c0e137ffe00f61225737184ece286d90cb5 (diff) |
Use memory statistics from model probing in calculation
Diffstat (limited to 'config-model-api/src')
-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) {} |