From 09a19dcddd7b487978195cff73c8194cf6aebb27 Mon Sep 17 00:00:00 2001 From: cassanof Date: Wed, 14 May 2025 21:38:03 -0700 Subject: [PATCH] fix --- 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 74ceff5..c3d3e96 100644 --- a/deep_gemm/jit/runtime.py +++ b/deep_gemm/jit/runtime.py @@ -49,7 +49,7 @@ class Runtime: result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) assert result.returncode == 0 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}' # Load kernel from the library