diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-05-08 13:54:37 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-05-08 13:54:37 +0200 |
commit | 85289c1c179d3469bfe7681ad3d04488185e6c7d (patch) | |
tree | 60f0230128cd5a9385299ca814abc960392ccd9e /config-model/src/test/derived | |
parent | a7abadf23150438e9b84577616698cf2605f6574 (diff) |
Add config for number of GPUs available to ONNX runtime
Diffstat (limited to 'config-model/src/test/derived')
3 files changed, 3 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 index d63e85e2f19..99f65336794 100644 --- a/config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg +++ b/config-model/src/test/derived/globalphase_onnx_inside/onnx-models.cfg @@ -1,3 +1,4 @@ +gpu.count -1 model[].name "direct" model[].fileref "files/ax_plus_b.onnx" model[].input[].name "vector_B" 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 index 6283159c324..cea4c065014 100644 --- a/config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg +++ b/config-model/src/test/derived/globalphase_token_functions/onnx-models.cfg @@ -1,3 +1,4 @@ +gpu.count -1 model[].name "my_ranking_model" model[].fileref "files/ranking_model.onnx" model[].input[].name "input_ids" diff --git a/config-model/src/test/derived/vector_constant/onnx-models.cfg b/config-model/src/test/derived/vector_constant/onnx-models.cfg index 4c52b72b519..1dcaf0e1bd6 100644 --- a/config-model/src/test/derived/vector_constant/onnx-models.cfg +++ b/config-model/src/test/derived/vector_constant/onnx-models.cfg @@ -1,3 +1,4 @@ +gpu.count -1 model[].name "inside" model[].fileref "ax_plus_b.onnx" model[].input[].name "vector_B" |