Make Kokkos configuration compatible with RTX40x0 generation consumer GPUs

@stanmoore1 is this the correct way add this?
This commit is contained in:
Axel Kohlmeyer
2023-02-07 08:30:30 -05:00
parent b9981ac51e
commit bde2867251
4 changed files with 8 additions and 1 deletions

View File

@ -74,6 +74,7 @@ int main() {
case 75: std::cout << "Set -DKokkos_ARCH_TURING75=ON ." << std::endl; break;
case 80: std::cout << "Set -DKokkos_ARCH_AMPERE80=ON ." << std::endl; break;
case 86: std::cout << "Set -DKokkos_ARCH_AMPERE86=ON ." << std::endl; break;
case 89: std::cout << "Set -DKokkos_ARCH_ADA89=ON ." << std::endl; break;
case 90: std::cout << "Set -DKokkos_ARCH_HOPPER90=ON ." << std::endl; break;
default:
std::cout << "Compute capability " << compute_capability