aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration/src
diff options
context:
space:
mode:
authorMartin Polden <mpolden@mpolden.no>2023-02-09 12:11:21 +0100
committerMartin Polden <mpolden@mpolden.no>2023-02-09 12:12:20 +0100
commite02adcebeb90661046893a2c386e330cdc925327 (patch)
tree317427d7f6ed1af49205017670ac87fc2c2089b3 /model-integration/src
parent5c5a79813ea6d6d57326f82145816200b8a41622 (diff)
Fix GPU detection
Diffstat (limited to 'model-integration/src')
-rw-r--r--model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java
index 563ef911f8f..5cc7991d197 100644
--- a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java
+++ b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java
@@ -101,7 +101,7 @@ public class OnnxEvaluator {
return createSession(modelPath, environment, options, false);
}
if (isCudaError(e)) {
- throw new IllegalArgumentException("GPU device is requested, but CUDA initialization failed", e);
+ throw new IllegalArgumentException("GPU device is required, but CUDA initialization failed", e);
}
throw new RuntimeException("ONNX Runtime exception", e);
}