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

Driver: fixed up invalid option for nvrtc

Browse Source
This commit is contained in:
Philippe Tillet
2015-08-26 13:44:40 -04:00
parent 0d3fcb18dc
commit 9da87bee51
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/driver/program.cpp
View File

@@ -44,7 +44,7 @@ Program::Program(Context const & context, std::string const & source) : backend_
nvrtc::check(dispatch::nvrtcCreateProgram(&prog, source.c_str(), NULL, 1, src, includes));
try{
std::pair<unsigned int, unsigned int> capability = context_.device().nv_compute_capability();
std::string capability_opt = "--compute_";
std::string capability_opt = "--gpu-architecture=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));