aboutsummaryrefslogtreecommitdiffstats
path: root/config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2021-09-30 16:05:03 +0200
committerLester Solbakken <lesters@oath.com>2021-09-30 16:05:03 +0200
commit387da217a9eb2a6f88b50f3608659a7d75c66aeb (patch)
treefbc82d6cb6add5d599425191cfc994c454fea4a6 /config-model/src/test/java/com/yahoo/vespa/model/ml/StatelessOnnxEvaluationTest.java
parent2f5a11f868291b34a3aa2c28817b36c5d0ed3d52 (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.java4
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());
}
}