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

[DRIVER] Now always using PTXv6.4

Browse Source
This commit is contained in:
Philippe Tillet
2020-07-29 13:50:01 -04:00
committed by Philippe Tillet
parent c251dc50f3
commit 29a0ad6c4d
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/driver/module.cc
View File

@@ -251,11 +251,10 @@ std::string cu_module::compile_llvm_module(std::unique_ptr<llvm::Module> module,
int version; int version;
dispatch::cuDriverGetVersion(&version); dispatch::cuDriverGetVersion(&version);
int major = version / 1000; int major = version / 1000;
int minor = (version - major*1000) / 10; // int minor = (version - major*1000) / 10;
if(major < 10) if(major < 10)
throw std::runtime_error("Triton requires CUDA 10+"); throw std::runtime_error("Triton requires CUDA 10+");
if(minor >= 1) find_and_replace(result, ".version", "\n", ".version 6.4\n");
find_and_replace(result, ".version", "\n", ".version 6.4\n");
while(find_and_replace(result, "\t// begin inline asm", "\n", "")); while(find_and_replace(result, "\t// begin inline asm", "\n", ""));
while(find_and_replace(result, "\t// end inline asm", "\n", "")); while(find_and_replace(result, "\t// end inline asm", "\n", ""));
return result; return result;