From 104a6ec109d8fe5e39141d19639c4f2067f506fb Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Thu, 15 May 2025 17:04:21 +0800 Subject: [PATCH] Add `__assertfail` --- deep_gemm/jit/runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_gemm/jit/runtime.py b/deep_gemm/jit/runtime.py index 52af8e1..3646d26 100644 --- a/deep_gemm/jit/runtime.py +++ b/deep_gemm/jit/runtime.py @@ -48,7 +48,7 @@ class Runtime: command = [f'{CUDA_HOME}/bin/cuobjdump', '-symbols', path] result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) 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]) kernel_names = [line.split()[-1] for line in result.stdout.splitlines() if line.startswith('STT_FUNC') and not check_illegal(line)]