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

Driver: now using proper compute capability option in nvrtc ; added missing file.

Browse Source
This commit is contained in:
Philippe Tillet
2015-08-26 13:31:58 -04:00
parent 0ce345f14a
commit 0d3fcb18dc
2 changed files with 1275 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/driver/program.cpp
View File

@@ -9,6 +9,7 @@
#include "sha1.hpp"
#include "to_string.hpp"
namespace isaac
{
@@ -42,7 +43,10 @@ Program::Program(Context const & context, std::string const & source) : backend_
nvrtc::check(dispatch::nvrtcCreateProgram(&prog, source.c_str(), NULL, 1, src, includes));
try{
const char * options[] = {"--gpu-architecture=compute_52", "--restrict"};
std::pair<unsigned int, unsigned int> capability = context_.device().nv_compute_capability();
std::string capability_opt = "--compute_";
capability_opt += tools::to_string(capability.first) + tools::to_string(capability.second);
const char * options[] = {capability_opt.c_str(), "--restrict"};
nvrtc::check(dispatch::nvrtcCompileProgram(prog, 2, options));
}catch(nvrtc::exception::compilation const &)
{