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

Merge branch 'rocm52_fixes_IFU' into fix_test_dot

Browse Source
This commit is contained in:
Michael Melesse
2022-10-31 19:24:45 +00:00
parent 277b712284 fecc7ce248
commit 4f3e2d6ed7
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lib/driver/llvm.cc
View File

@@ -26,6 +26,7 @@
#include <memory> #include <memory>
#include <regex> #include <regex>
#include <iomanip> #include <iomanip>
#include <string>
#include "triton/driver/llvm.h" #include "triton/driver/llvm.h"
#include "triton/driver/dispatch.h" #include "triton/driver/dispatch.h"
#include "triton/driver/error.h" #include "triton/driver/error.h"
@@ -70,6 +71,24 @@ extern "C"
int setupterm(char *term, int fildes, int *errret) { return 0; } int setupterm(char *term, int fildes, int *errret) { return 0; }
} }
namespace {
std::string gen_random(const int len) {
static const char alphanum[] =
"0123456789"
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"abcdefghijklmnopqrstuvwxyz";
std::string tmp_s;
tmp_s.reserve(len);
for (int i = 0; i < len; ++i) {
tmp_s += alphanum[rand() % (sizeof(alphanum) - 1)];
}
return tmp_s;
}
}
namespace triton namespace triton
{ {
namespace driver namespace driver
@@ -286,7 +305,7 @@ namespace triton
auto in_time_t = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now()); auto in_time_t = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
std::stringstream cur_time; std::stringstream cur_time;
cur_time << std::put_time(std::localtime(&in_time_t), "%Y-%m-%d--%I-%M-%S"); cur_time << std::put_time(std::localtime(&in_time_t), "%Y-%m-%d--%I-%M-%S");
std::string kernel_name = module->getModuleIdentifier() + "_" + cur_time.str(); std::string kernel_name = module->getModuleIdentifier() + "_" + cur_time.str() + "_" + gen_random(12);
// verify and store llvm // verify and store llvm
llvm::legacy::PassManager pm; llvm::legacy::PassManager pm;
pm.add(llvm::createVerifierPass()); pm.add(llvm::createVerifierPass());