Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[onnxruntime] Add debug message for OnnxRuntime #3217

Merged
merged 1 commit into from
May 24, 2024

Conversation

xyang16
Copy link
Contributor

@xyang16 xyang16 commented May 24, 2024

Description

Brief description of what this PR is about

  • If this change is a backward incompatible change, why must this change be made?
  • Interesting edge cases to note here

@xyang16 xyang16 requested review from zachgk, frankfliu and a team as code owners May 24, 2024 22:55
@@ -110,6 +114,7 @@ public boolean hasCapability(String capability) {
sessionOptions.addCUDA();
return true;
} catch (OrtException e) {
logger.warn("Failed to add CUDA for OnnxRuntime engine: " + e.getMessage());
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
logger.warn("Failed to add CUDA for OnnxRuntime engine: " + e.getMessage());
logger.warn("Failed to add CUDA for OnnxRuntime engine", e);

Copy link
Contributor Author

@xyang16 xyang16 May 24, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is just a warning to help debug the reason why CUDA not supported, we don't need the whole stacktrace, that's why I think only print the error message should be enough.

@xyang16 xyang16 merged commit c85040e into deepjavalibrary:master May 24, 2024
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants