diff options
author | Jo Kristian Bergum <bergum@yahooinc.com> | 2023-10-26 09:31:18 +0200 |
---|---|---|
committer | Jo Kristian Bergum <bergum@yahooinc.com> | 2023-10-26 09:31:18 +0200 |
commit | 8aea914273b599112047f262e7598c9b9dc2e05a (patch) | |
tree | a638ade5fc96fc7ac04ef330d6a9e840d96ef902 /model-integration/src/main/java/ai | |
parent | 4a81c971976f3b3051d741cc046183bc15910c58 (diff) |
Less verbose logging when failing to find CUDA and it is optional
Diffstat (limited to 'model-integration/src/main/java/ai')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java | 4 |
1 files changed, 2 insertions, 2 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 628fe933bf5..cd698eb1647 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 @@ -151,8 +151,8 @@ public class OnnxEvaluator implements AutoCloseable { throw new IllegalArgumentException("No such file: " + model.path().get()); } if (tryCuda && isCudaError(e) && !options.gpuDeviceRequired()) { - LOG.log(Level.WARNING, "Failed to create session with CUDA using GPU device " + - options.gpuDeviceNumber() + ". Falling back to CPU", e); + LOG.log(Level.INFO, "Failed to create session with CUDA using GPU device " + + options.gpuDeviceNumber() + ". Falling back to CPU. Reason: " + e.getMessage()); return createSession(model, runtime, options, false); } if (isCudaError(e)) { |