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

[RUNTIME] removed obsolete putenv call (#305)

Browse Source
This commit is contained in:
Philippe Tillet
2021-09-23 17:51:58 -07:00
committed by GitHub
parent 87f8d9f163
commit e22d92c63c
1 changed files with 5 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lib/driver/dispatch.cc
View File

@@ -21,7 +21,6 @@
*/ */
#include "triton/driver/dispatch.h" #include "triton/driver/dispatch.h"
#include "triton/tools/sys/getenv.hpp"
namespace triton namespace triton
{ {
@@ -92,17 +91,11 @@ void* dispatch::fname ## _;
bool dispatch::cuinit(){ bool dispatch::cuinit(){
if(cuda_==nullptr){ if(cuda_==nullptr){
putenv((char*)"CUDA_CACHE_DISABLE=1"); cuda_ = dlopen("libcuda.so", RTLD_LAZY);
std::string libcuda = tools::getenv("TRITON_LIBCUDA"); if(!cuda_)
if(libcuda.empty()){ cuda_ = dlopen("libcuda.so.1", RTLD_LAZY);
cuda_ = dlopen("libcuda.so", RTLD_LAZY); if(!cuda_)
if(!cuda_) throw std::runtime_error("Could not find `libcuda.so`. Make sure it is in your LD_LIBRARY_PATH.");
cuda_ = dlopen("libcuda.so.1", RTLD_LAZY);
if(!cuda_)
throw std::runtime_error("Could not find `libcuda.so`. Make sure it is in your LD_LIBRARY_PATH.");
}
else
cuda_ = dlopen(libcuda.c_str(), RTLD_LAZY);
} }
if(cuda_ == nullptr) if(cuda_ == nullptr)
return false; return false;