diff options
author | Martin Polden <mpolden@mpolden.no> | 2023-01-23 14:34:35 +0100 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2023-01-23 14:34:35 +0100 |
commit | c9ba67dd5b8402125ea84a5c5fd12562ca7ebd15 (patch) | |
tree | 415a4d8ba38f8a9b52cd158f1540d19cf337bccf /searchcore | |
parent | 4ffc7ef27da2a8d824c9f04a80f89e569e36b322 (diff) |
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 |