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

fix

Browse Source
This commit is contained in:
cassanof 2025-05-14 21:38:03 -07:00
parent e2d6a107ef
commit 09a19dcddd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deep_gemm/jit/runtime.py
View File

@ -49,7 +49,7 @@ class Runtime:
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
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 '__instantiate_kernel' not in line] if line.startswith('STT_FUNC') and '__instantiate_kernel' not in line and '__assertfail' not in line]
assert len(kernel_names) == 1, f'Too many kernels in the library: {kernel_names}' assert len(kernel_names) == 1, f'Too many kernels in the library: {kernel_names}'
# Load kernel from the library # Load kernel from the library