gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Philippe Tillet
2021-07-29 11:27:55 -07:00
parent 41ecd96300
commit 2f0f51be50
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/driver/module.cc
View File

@@ -242,7 +242,7 @@ std::string cu_module::compile_llvm_module(llvm::Module* module, driver::device*
if(major < 10) if(major < 10)
throw std::runtime_error("Triton requires CUDA 10+"); throw std::runtime_error("Triton requires CUDA 10+");
// PTX version // PTX version
int ptx = vptx.at(version); int ptx = version > 11040 ? 73 : vptx.at(version);
int ptx_major = ptx / 10; int ptx_major = ptx / 10;
int ptx_minor = ptx % 10; int ptx_minor = ptx % 10;
// create // create