summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorArne Juul <arnej@yahooinc.com>2023-03-22 15:01:13 +0000
committerArne Juul <arnej@yahooinc.com>2023-03-22 15:03:41 +0000
commita06c1cf91899f5da327c408c61c798ffddfd32da (patch)
tree80acab4d624c214f04005aa194033c3f6d499b55
parent39f26bc5834d54d4dad657866cf6f74e62b9cf95 (diff)
add testing of derived onnx-models.cfg
-rw-r--r--config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg64
-rw-r--r--config-model/src/test/derived/globalphase_onnx_inside/test.sd3
-rw-r--r--config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg16
-rw-r--r--config-model/src/test/derived/vector_constant/onnx-models.cfg16
4 files changed, 99 insertions, 0 deletions
diff --git a/config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg b/config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg
new file mode 100644
index 00000000000..f705f896540
--- /dev/null
+++ b/config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg
@@ -0,0 +1,64 @@
+model[].name "direct"
+model[].fileref "files/ax_plus_b.onnx"
+model[].input[].name "vector_B"
+model[].input[].source "query(bb)"
+model[].input[].name "vector_A"
+model[].input[].source "attribute(aa)"
+model[].input[].name "matrix_X"
+model[].input[].source "constant(xx)"
+model[].output[].name "vector_Y"
+model[].output[].as "out"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads -1
+model[].stateless_intraop_threads -1
+model[].gpu_device -1
+model[].gpu_device_required false
+model[].name "inside"
+model[].fileref "files/ax_plus_b.onnx"
+model[].input[].name "vector_B"
+model[].input[].source "rankingExpression(indirect_b)"
+model[].input[].name "vector_A"
+model[].input[].source "rankingExpression(indirect_a)"
+model[].input[].name "matrix_X"
+model[].input[].source "rankingExpression(indirect_x)"
+model[].output[].name "vector_Y"
+model[].output[].as "foobar"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads 5
+model[].stateless_intraop_threads 3
+model[].gpu_device 2
+model[].gpu_device_required false
+model[].name "twoside"
+model[].fileref "files/ax_plus_b.onnx"
+model[].input[].name "vector_B"
+model[].input[].source "rankingExpression(indirect_b)"
+model[].input[].name "vector_A"
+model[].input[].source "rankingExpression(indirect_a)"
+model[].input[].name "matrix_X"
+model[].input[].source "rankingExpression(indirect_x)"
+model[].output[].name "vector_Y"
+model[].output[].as "foobar"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads -1
+model[].stateless_intraop_threads -1
+model[].gpu_device -1
+model[].gpu_device_required false
+model[].name "another"
+model[].fileref "files/ax_plus_b.onnx"
+model[].input[].name "vector_B"
+model[].input[].source "rankingExpression(indirect_b)"
+model[].input[].name "vector_A"
+model[].input[].source "rankingExpression(indirect_a)"
+model[].input[].name "matrix_X"
+model[].input[].source "rankingExpression(indirect_x)"
+model[].output[].name "vector_Y"
+model[].output[].as "foobar"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads -1
+model[].stateless_intraop_threads -1
+model[].gpu_device -1
+model[].gpu_device_required false
diff --git a/config-model/src/test/derived/globalphase_onnx_inside/test.sd b/config-model/src/test/derived/globalphase_onnx_inside/test.sd
index 3e48f912931..cd3dfcea3d0 100644
--- a/config-model/src/test/derived/globalphase_onnx_inside/test.sd
+++ b/config-model/src/test/derived/globalphase_onnx_inside/test.sd
@@ -49,6 +49,9 @@ schema test {
input matrix_X: indirect_x
input vector_B: indirect_b
output vector_Y: foobar
+ intraop-threads: 3
+ interop-threads: 5
+ gpu-device: 2
}
first-phase {
expression: sum(attribute(aa))
diff --git a/config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg b/config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg
new file mode 100644
index 00000000000..6283159c324
--- /dev/null
+++ b/config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg
@@ -0,0 +1,16 @@
+model[].name "my_ranking_model"
+model[].fileref "files/ranking_model.onnx"
+model[].input[].name "input_ids"
+model[].input[].source "rankingExpression(input_ids)"
+model[].input[].name "attention_mask"
+model[].input[].source "rankingExpression(attention_mask)"
+model[].input[].name "token_type_ids"
+model[].input[].source "rankingExpression(token_type_ids)"
+model[].output[].name "score"
+model[].output[].as "score"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads -1
+model[].stateless_intraop_threads -1
+model[].gpu_device -1
+model[].gpu_device_required false
diff --git a/config-model/src/test/derived/vector_constant/onnx-models.cfg b/config-model/src/test/derived/vector_constant/onnx-models.cfg
new file mode 100644
index 00000000000..4c52b72b519
--- /dev/null
+++ b/config-model/src/test/derived/vector_constant/onnx-models.cfg
@@ -0,0 +1,16 @@
+model[].name "inside"
+model[].fileref "ax_plus_b.onnx"
+model[].input[].name "vector_B"
+model[].input[].source "constant(bb)"
+model[].input[].name "vector_A"
+model[].input[].source "rankingExpression(indirect_a)"
+model[].input[].name "matrix_X"
+model[].input[].source "constant(xx)"
+model[].output[].name "vector_Y"
+model[].output[].as "foobar"
+model[].dry_run_on_setup true
+model[].stateless_execution_mode ""
+model[].stateless_interop_threads -1
+model[].stateless_intraop_threads -1
+model[].gpu_device -1
+model[].gpu_device_required false