Skip to content

Commit

Permalink
[Fix] Fix plthook无法hook dlopen 动态加载的so
Browse files Browse the repository at this point in the history
  • Loading branch information
LamForest committed Dec 18, 2024
1 parent 9a2e5e9 commit b34427d
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion src/cuda_mock/cuda_mock_impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,8 +141,14 @@ def dump_to_cache(self):
]
class __XpuRuntimeProfiler:
def __init__(self, target_libs = gDefaultTargetLib, target_symbols = gDefaultTargetSymbols):
print_hook_initialize(target_libs=target_libs, target_symbols=target_symbols)
self.target_libs = target_libs
self.target_symbols = target_symbols
self.init = False
def start_capture(self):
# Note: 延迟plthook到第一次使用,此时更多的so已经加载完成(比如dlopen打开的so)
if not self.init:
print_hook_initialize(target_libs=self.target_libs, target_symbols=self.target_symbols)
self.init = True
cuda_mock_impl.print_hook_start_capture()

def end_capture(self, op_key):
Expand Down

0 comments on commit b34427d

Please sign in to comment.