diff options
Diffstat (limited to 'model-integration')
-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); } |