-
Notifications
You must be signed in to change notification settings - Fork 407
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
Add configuration to target CUDA compute capability 8.6 #3713
Conversation
Can one of the admins verify this patch? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me.
OK to test |
I ran the unit tests on RTX3090 and the following tests in kokkos/core/unit_test/TestTeamTeamSize.hpp Lines 163 to 167 in 78538c8
kokkos/core/unit_test/TestTeamTeamSize.hpp Lines 178 to 182 in 78538c8
After some debugging I figured out |
Changes were added
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
This PR adds a flag to compile Kokkos with CUDA compute capability 8.6 (changes are based on #3122). I need this because I'm working with Ampere-based RTX cards.