aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorMartin Polden <mpolden@mpolden.no>2023-01-24 14:56:16 +0100
committerMartin Polden <mpolden@mpolden.no>2023-01-24 14:57:06 +0100
commit5e587acf9ed220327c12b1949869ea414af89e13 (patch)
tree2d868a0a9c9f1f98eccf9039cf6c8d327b554b49 /model-integration
parent16b5b25f8c7512c122bb3e7c1cd5c412f8506c48 (diff)
Do not filter on error code
Diffstat (limited to 'model-integration')
-rw-r--r--model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorOptions.java5
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);
}