diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-03-22 15:01:13 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-03-22 15:03:41 +0000 |
commit | a06c1cf91899f5da327c408c61c798ffddfd32da (patch) | |
tree | 80acab4d624c214f04005aa194033c3f6d499b55 | |
parent | 39f26bc5834d54d4dad657866cf6f74e62b9cf95 (diff) |
add testing of derived onnx-models.cfg
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 |