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.

Sign up for GitHub

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

Jump to bottom

Check DLRM cuda files can run on GPUs newer than Ampere #165

Open
fotstrt opened this issue Feb 27, 2023 · 0 comments
Open

Check DLRM cuda files can run on GPUs newer than Ampere #165

fotstrt opened this issue Feb 27, 2023 · 0 comments
Labels
Models

Comments

@fotstrt
Copy link
Contributor

fotstrt commented Feb 27, 2023

Currently, we know that the DLRM cuda libraries written by NVIDIA can run on V100 (compute capability 7.0) and A100 GPUs (compute capability 8.0). If we move to a newer GPU, such as RTX 3090 (compute capability 8.6), would the existing code work there? We might also need to change the nvcc flags for compilation.
@fotstrt fotstrt added the Models label Feb 28, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
Models
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@fotstrt