Skip to content

add warning to the nvjitlink ctor when falling back to cuLink #315

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

Merged
merged 1 commit into from
Jan 2, 2025

Conversation

ksimpson-work
Copy link
Contributor

@ksimpson-work ksimpson-work commented Dec 19, 2024

close #303

Add a warning when we fall back to cuLink

Copy link
Contributor

copy-pr-bot bot commented Dec 19, 2024

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@ksimpson-work ksimpson-work self-assigned this Dec 19, 2024
@ksimpson-work ksimpson-work added enhancement Any code-related improvements cuda.core Everything related to the cuda.core module labels Dec 19, 2024
@leofang leofang linked an issue Dec 20, 2024 that may be closed by this pull request
@leofang leofang added this to the cuda.core beta 3 milestone Dec 20, 2024
@leofang leofang added the P1 Medium priority - Should do label Dec 26, 2024
@ksimpson-work
Copy link
Contributor Author

/ok to test

@ksimpson-work ksimpson-work marked this pull request as ready for review January 1, 2025 22:00
@ksimpson-work ksimpson-work requested a review from leofang January 1, 2025 22:00
Copy link
Member

@leofang leofang left a comment

Choose a reason for hiding this comment

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

Thanks, Keenan, LGTM!

warnings.warn(
"nvJitLink is not installed or too old (<12.3). Therefore it is not usable "
"and the culink APIs will be used instead.",
stacklevel=3,
Copy link
Member

Choose a reason for hiding this comment

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

Q: Is stacklevel=2 not enough? I always forget which level is the right choice, but I'd prefer to not print out the cuda.core internal as part of the traceback.

@ksimpson-work ksimpson-work merged commit 5b98fdc into NVIDIA:main Jan 2, 2025
43 of 46 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cuda.core Everything related to the cuda.core module enhancement Any code-related improvements P1 Medium priority - Should do
2 participants