summaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
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);
}