diff options
author | Lester Solbakken <lesters@oath.com> | 2021-09-30 16:05:03 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-09-30 16:05:03 +0200 |
commit | 387da217a9eb2a6f88b50f3608659a7d75c66aeb (patch) | |
tree | fbc82d6cb6add5d599425191cfc994c454fea4a6 /config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java | |
parent | 2f5a11f868291b34a3aa2c28817b36c5d0ed3d52 (diff) |
Add parsing of ONNX Runtime session options to services.xml
Diffstat (limited to 'config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java')
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java b/config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java index b369560be74..27f1fc26b15 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java @@ -123,6 +123,10 @@ public class StatelessOnnxEvaluationTest { Tensor output = evaluator.bind("input1", input1).bind("input2", input2).evaluate(); assertEquals(6.0, output.sum().asDouble(), 1e-9); + OnnxModelsConfig.Model mulModel = onnxModelsConfig.model().get(0); + assertEquals(2, mulModel.stateless_intraop_threads()); + assertEquals(-1, mulModel.stateless_interop_threads()); + assertEquals("", mulModel.stateless_execution_mode()); } } |