chenchenghao/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix CUBIN symbol name compatibility

Browse Source
This commit is contained in:
Chenggang Zhao
2025-08-26 17:43:26 +08:00
parent 89b4089d24
commit 9c3783beb2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
csrc/jit/kernel_runtime.hpp
+2 -1
View File
@@ -46,7 +46,8 @@ public:
std::istringstream iss(symbols); std::istringstream iss(symbols);
std::vector<std::string> symbol_names; std::vector<std::string> symbol_names;
for (std::string line; std::getline(iss, line); ) { for (std::string line; std::getline(iss, line); ) {
if (line.find("STT_FUNC") == 0 and std::none_of(illegal_names.begin(), illegal_names.end(), if (line.find("STT_FUNC") == 0 and line.find("STO_ENTRY") != std::string::npos and
std::none_of(illegal_names.begin(), illegal_names.end(),
[&](const auto& name) { return line.find(name) != std::string::npos; })) { [&](const auto& name) { return line.find(name) != std::string::npos; })) {
const auto& last_space = line.rfind(' '); const auto& last_space = line.rfind(' ');
symbol_names.push_back(line.substr(last_space + 1)); symbol_names.push_back(line.substr(last_space + 1));