Cuda & PTX version updates.

tarunprabhu · Jul 25, 2024 · e5876fd · e5876fd
1 parent 44b75a0
commit e5876fd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/llvm/lib/Transforms/Tapir/CudaABI.cpp b/llvm/lib/Transforms/Tapir/CudaABI.cpp
@@ -310,10 +310,12 @@ std::string PTXVersionFromCudaVersion() {
           .Case("11.6", "+ptx76")
           .Case("11.7", "+ptx77")
           .Case("11.8", "+ptx78")
-          .Case("12.0", "+ptx78")
-          .Case("12.1", "+ptx78")
-          .Case("12.2", "+ptx78")
-          .Case("12.3", "+ptx78")
+          .Case("12.0", "+ptx83")
+          .Case("12.1", "+ptx83")
+          .Case("12.2", "+ptx83")
+          .Case("12.3", "+ptx83")
+          .Case("12.4", "+ptx83")
+          .Case("12.5", "+ptx83")    
           .Default("");
 
   if (PTXVersionStr == "") {