diff options
author | Martin Polden <mpolden@mpolden.no> | 2023-01-24 10:20:07 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-01-24 10:20:07 +0100 |
commit | c0c2437269f95a77cc2f5e25c5c8e077f632f1fa (patch) | |
tree | 8a3520f0c32d8704786c47f7cc9af7fe3d0477ec /searchcore | |
parent | f31bf6e1c3482d785662700ba2ef8f5a3bd2443e (diff) | |
parent | c9ba67dd5b8402125ea84a5c5fd12562ca7ebd15 (diff) |
Merge pull request #25680 from vespa-engine/mpolden/set-gpu-device
Support configuration of GPU device to use in ONNX model
Diffstat (limited to 'searchcore')
-rw-r--r-- | searchcore/src/vespa/searchcore/config/onnx-models.def | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/searchcore/src/vespa/searchcore/config/onnx-models.def b/searchcore/src/vespa/searchcore/config/onnx-models.def index c117d6f162c..b8f5d319075 100644 --- a/searchcore/src/vespa/searchcore/config/onnx-models.def +++ b/searchcore/src/vespa/searchcore/config/onnx-models.def @@ -11,3 +11,5 @@ model[].dry_run_on_setup bool default=false model[].stateless_execution_mode string default="" model[].stateless_interop_threads int default=-1 model[].stateless_intraop_threads int default=-1 +model[].gpu_device int default=-1 +model[].gpu_device_required bool default=false |