DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add __assertfail

Browse Source
This commit is contained in:
Chenggang Zhao
2025-05-15 17:04:21 +08:00
parent 3b412f458a
commit 104a6ec109
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
deep_gemm/jit/runtime.py
View File

@@ -48,7 +48,7 @@ class Runtime:
command = [f'{CUDA_HOME}/bin/cuobjdump', '-symbols', path] command = [f'{CUDA_HOME}/bin/cuobjdump', '-symbols', path]
result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
assert result.returncode == 0 assert result.returncode == 0
illegal_names = ['vprintf', '__instantiate_kernel', '__internal'] illegal_names = ['vprintf', '__instantiate_kernel', '__internal', '__assertfail']
check_illegal = lambda line: any([name in line for name in illegal_names]) check_illegal = lambda line: any([name in line for name in illegal_names])
kernel_names = [line.split()[-1] for line in result.stdout.splitlines() kernel_names = [line.split()[-1] for line in result.stdout.splitlines()
if line.startswith('STT_FUNC') and not check_illegal(line)] if line.startswith('STT_FUNC') and not check_illegal(line)]