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 /config-model/src/main/resources/schema/containercluster.rnc | |
parent | 4ffc7ef27da2a8d824c9f04a80f89e569e36b322 (diff) |
Support configuration of GPU device to use in ONNX model
Diffstat (limited to 'config-model/src/main/resources/schema/containercluster.rnc')
-rw-r--r-- | config-model/src/main/resources/schema/containercluster.rnc | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/config-model/src/main/resources/schema/containercluster.rnc b/config-model/src/main/resources/schema/containercluster.rnc index 81455084ad2..888851b2db2 100644 --- a/config-model/src/main/resources/schema/containercluster.rnc +++ b/config-model/src/main/resources/schema/containercluster.rnc @@ -103,7 +103,11 @@ ModelEvaluation = element model-evaluation { attribute name { string } & element intraop-threads { xsd:nonNegativeInteger }? & element interop-threads { xsd:nonNegativeInteger }? & - element execution-mode { string "sequential" | string "parallel" }? + element execution-mode { string "sequential" | string "parallel" }? & + element gpu-device { + attribute required { xsd:boolean } & + xsd:nonNegativeInteger + }? }* }? }? |