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

[RUNTIME] Disable error on spills

Browse Source
This commit is contained in:
Philippe Tillet
2021-01-14 20:33:34 -05:00
parent 3b36a1e60c
commit 376c876eb8
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/driver/module.cc
View File

@@ -330,15 +330,13 @@ cu_module::cu_module(driver::device* device, std::string const & source) : modul
dispatch::cuModuleLoadDataEx(&*cu_, ptx_.data(), 5, opt, optval);
std::string err(_err);
std::string log(_log);
// std::cout << log << std::endl;
std::smatch match;
std::regex expr ("\\b([0-9]+) bytes spill");
spilled_ = 0;
while (std::regex_search(log,match,expr)){
spilled_ += std::stoi(match[1]);
log = match.suffix();
}
// std::smatch match;
// std::regex expr ("\\b([0-9]+) bytes spill");
// spilled_ = 0;
// while (std::regex_search(log,match,expr)){
// spilled_ += std::stoi(match[1]);
// log = match.suffix();
// }
}
catch(exception::cuda::invalid_ptx const &){
//#ifdef TRITON_LOG_PTX_ERROR

4
lib/runtime/function.cc
View File

@@ -251,8 +251,8 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::d
// ir::print(module, std::cout);
isel.visit(module, *llvm);
std::unique_ptr<driver::module> res(driver::module::create(device, std::move(llvm)));
if(res->spilled() > 256)
throw exception::out_of_registers();
// if(res->spilled() > 256)
// throw exception::out_of_registers();
return res;
}