diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-03-22 14:59:59 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-03-22 15:03:41 +0000 |
commit | 5f3c9493cf4049c27b0a9284f5a49918084fb651 (patch) | |
tree | fee24eb7d227284e66e09271fdf1985695ebfcfa /config-model | |
parent | 5a78d21b77d9dc70896182a83d29e855214514b3 (diff) |
extend grammar with new settings
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/javacc/SchemaParser.jj | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/config-model/src/main/javacc/SchemaParser.jj b/config-model/src/main/javacc/SchemaParser.jj index fa9d34139ea..61e8574bc87 100644 --- a/config-model/src/main/javacc/SchemaParser.jj +++ b/config-model/src/main/javacc/SchemaParser.jj @@ -186,6 +186,9 @@ TOKEN : | < SUFFIX: "suffix" > | < CONSTANT: "constant"> | < ONNXMODEL: "onnx-model"> +| < INTRAOPTHREADS: "intraop-threads"> +| < INTEROPTHREADS: "interop-threads"> +| < GPUDEVICE: "gpu-device"> | < MODEL: "model" > | < MUTATE: "mutate" > | < QUERY: "query" > @@ -1594,11 +1597,15 @@ OnnxModel onnxModel() : void onnxModelItem(OnnxModel onnxModel) : { String path = null; + int num; } { ( (path = fileItem()) { onnxModel.setFileName(path); } | (path = uriItem()) { onnxModel.setUri(path); } | + <GPUDEVICE> <COLON> num = integer() { onnxModel.setGpuDevice(num, false); } | + <INTRAOPTHREADS> <COLON> num = integer() { onnxModel.setStatelessIntraOpThreads(num); } | + <INTEROPTHREADS> <COLON> num = integer() { onnxModel.setStatelessInterOpThreads(num); } | (<ONNX_INPUT_SL>) { String name = token.image.substring(5, token.image.lastIndexOf(":")).trim(); if (name.startsWith("\"")) { name = name.substring(1, name.length() - 1); } |