豆豆友情提示:这是一个非官方 GitHub 代理镜像,主要用于网络测试或访问加速。请勿在此进行登录、注册或处理任何敏感信息。进行这些操作请务必访问官方网站 github.com。 Raw 内容也通过此代理提供。
Skip to content

Commit 2f58e70

Browse files
committed
Fix CUBIN symbol name compatibility
1 parent a3d1aa4 commit 2f58e70

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

csrc/jit/kernel_runtime.hpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,8 @@ class KernelRuntime final {
4646
std::istringstream iss(symbols);
4747
std::vector<std::string> symbol_names;
4848
for (std::string line; std::getline(iss, line); ) {
49-
if (line.find("STT_FUNC") == 0 and std::none_of(illegal_names.begin(), illegal_names.end(),
49+
if (line.find("STT_FUNC") == 0 and line.find("STO_ENTRY") != std::string::npos and
50+
std::none_of(illegal_names.begin(), illegal_names.end(),
5051
[&](const auto& name) { return line.find(name) != std::string::npos; })) {
5152
const auto& last_space = line.rfind(' ');
5253
symbol_names.push_back(line.substr(last_space + 1));

0 commit comments

Comments
 (0)