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

[DRIVER] Now giving the option to use system ptxas through environment variable (#123)

Browse Source
This commit is contained in:
Philippe Tillet
2021-06-11 13:48:11 -04:00
committed by Philippe Tillet
parent 80c86ecf4a
commit b7b05a560e
3 changed files with 40 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/triton/driver/module.h
View File

@@ -60,7 +60,7 @@ public:
// CUDA
class cu_module: public module {
std::string compile_llvm_module(llvm::Module* module, driver::device* device);
void init_from_ptx(const std::string& ptx);
void init_from_ptx(const std::string& ptx, cu_device *device);
public:
cu_module(driver::device* device, std::unique_ptr<llvm::Module> module);