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

[DRIVER] Better exception handling of invalid ptx

Browse Source
This commit is contained in:
Philippe Tillet
2020-11-27 17:00:11 -05:00
parent 44ca2c0cb8
commit 5e8f4c934c
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

@@ -276,7 +276,7 @@ cu_module::cu_module(std::string const & source) : module(CUmodule(), true), sou
void* optval[] = {(void*)(uintptr_t)errbufsize, (void*)errbuf.data()}; void* optval[] = {(void*)(uintptr_t)errbufsize, (void*)errbuf.data()};
try{ try{
dispatch::cuModuleLoadDataEx(&*cu_, source_.data(), 2, opt, optval); dispatch::cuModuleLoadDataEx(&*cu_, source_.data(), 2, opt, optval);
}catch(exception::cuda::base const &){ }catch(exception::cuda::invalid_ptx const &){
//#ifdef TRITON_LOG_PTX_ERROR //#ifdef TRITON_LOG_PTX_ERROR
std::cout << source << std::endl; std::cout << source << std::endl;
std::cerr << "It appears that Triton produced invalid PTX code:" << std::endl; std::cerr << "It appears that Triton produced invalid PTX code:" << std::endl;