Use ptx 8.5 with cuda-12.6, and enable ptx 8.6 with the newer CUDA ve…

…rsions
llvm · Oct 14, 2024 · aee63ae · aee63ae
1 parent 5dac14a
commit aee63ae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/clang/lib/Basic/Cuda.cpp b/clang/lib/Basic/Cuda.cpp
@@ -224,7 +224,7 @@ CudaVersion MinVersionForOffloadArch(OffloadArch A) {
   case OffloadArch::SM_90a:
     return CudaVersion::CUDA_120;
   case OffloadArch::SM_100:
-    return CudaVersion::CUDA_126;
+    return CudaVersion::NEW; // TODO: use specific CUDA version once it's public.
   default:
     llvm_unreachable("invalid enum");
   }

diff --git a/clang/lib/Driver/ToolChains/Cuda.cpp b/clang/lib/Driver/ToolChains/Cuda.cpp
@@ -671,7 +671,11 @@ void NVPTX::getNVPTXTargetFeatures(const Driver &D, const llvm::Triple &Triple,
   case CudaVersion::CUDA_##CUDA_VER:                                           \
     PtxFeature = "+ptx" #PTX_VER;                                              \
     break;
-    CASE_CUDA_VERSION(126, 86);
+    // TODO: Use specific CUDA version once it's public.
+    case clang::CudaVersion::NEW:
+      PtxFeature = "+ptx86";
+      break;
+    CASE_CUDA_VERSION(126, 85);
     CASE_CUDA_VERSION(125, 85);
     CASE_CUDA_VERSION(124, 84);
     CASE_CUDA_VERSION(123, 83);