[DRIVER] No longer crashing when encountering CUDA version >11.4

2021-07-29 11:27:55 -07:00
parent 41ecd96300
commit 2f0f51be50
1 changed files with 1 additions and 1 deletions
--- a/lib/driver/module.cc
+++ b/lib/driver/module.cc
@@ -242,7 +242,7 @@ std::string cu_module::compile_llvm_module(llvm::Module* module, driver::device*
  if(major < 10)
    throw std::runtime_error("Triton requires CUDA 10+");
  // PTX version
-  int ptx = vptx.at(version);
+  int ptx = version > 11040 ? 73 : vptx.at(version);
  int ptx_major = ptx / 10;
  int ptx_minor = ptx % 10;
  // create