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

Primerun: Ensure nvidia_icd.json is used #956

Merged
merged 2 commits into from Oct 23, 2023
Merged

Primerun: Ensure nvidia_icd.json is used #956

merged 2 commits into from Oct 23, 2023

Conversation

sonic2kk
Copy link
Owner

@sonic2kk sonic2kk commented Oct 23, 2023

Should fix #953.

When USEPRIMERUN is on, we export several environment variables. But if the nvidia_icd.json is not set in the current environment's VK_ICD_FILENAMES, it will not be able to use the Nvidia GPU. For cases likee this, we should explicitly export nvidia_icd.json.

The fix for this was given by OP in #953 (they informed which environment variable to use), this PR just applies it in the appropriate spot in the code.

This path is not likely to change, and if other custom ones are necessary, a user can manually set these in their STL env variable config file.

Putting this up as a branch in case this fix is not sufficient, and we have to make other changes.

@sonic2kk sonic2kk changed the title Primerun: Ensure nvidiaa_icd.json is used Primerun: Ensure nvidia_icd.json is used Oct 23, 2023
@sonic2kk
Copy link
Owner Author

OP confirmed it to work, shellcheck is green. Ready to merge.

@sonic2kk sonic2kk merged commit 87bb53a into master Oct 23, 2023
@sonic2kk sonic2kk deleted the primerun-change branch October 23, 2023 17:30
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.

primerun setting doesn't work
1 participant