From 2f0f51be50b97c4831c7b0f00064efa3956b817b Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Thu, 29 Jul 2021 11:27:55 -0700 Subject: [PATCH] [DRIVER] No longer crashing when encountering CUDA version >11.4 --- lib/driver/module.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/driver/module.cc b/lib/driver/module.cc index 3f5b1d953..ff8f37665 100755 --- 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