diff options
author | Martin Polden <mpolden@mpolden.no> | 2023-01-24 14:56:16 +0100 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2023-01-24 14:57:06 +0100 |
commit | 5e587acf9ed220327c12b1949869ea414af89e13 (patch) | |
tree | 2d868a0a9c9f1f98eccf9039cf6c8d327b554b49 /model-integration/src/main/java/ai | |
parent | 16b5b25f8c7512c122bb3e7c1cd5c412f8506c48 (diff) |
Do not filter on error code
Diffstat (limited to 'model-integration/src/main/java/ai')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java index fceb63e6ae6..cdbce760d92 100644 --- a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java +++ b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java @@ -39,14 +39,11 @@ public class OnnxEvaluatorOptions { return options; } - private void addCuda(OrtSession.SessionOptions options) throws OrtException { + private void addCuda(OrtSession.SessionOptions options) { if (gpuDeviceNumber < 0) return; try { options.addCUDA(gpuDeviceNumber); } catch (OrtException e) { - if (e.getCode() != OrtException.OrtErrorCode.ORT_EP_FAIL) { - throw e; - } if (gpuDeviceRequired) { throw new IllegalArgumentException("GPU device " + gpuDeviceNumber + " is required, but CUDA backend could not be initialized", e); } |