From 3f22f813265c70a1db36579c5b06e29cbd6cf46f Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Wed, 7 May 2025 11:30:57 +0800 Subject: [PATCH] Add a TODO --- deep_gemm/jit/runtime.py | 1 + 1 file changed, 1 insertion(+) diff --git a/deep_gemm/jit/runtime.py b/deep_gemm/jit/runtime.py index c8356a0..b7c2f95 100644 --- a/deep_gemm/jit/runtime.py +++ b/deep_gemm/jit/runtime.py @@ -44,6 +44,7 @@ class Runtime: assert result == cbd.CUresult.CUDA_SUCCESS, f'Failed to load library: {result}' # Extract the kernel name + # TODO: use `cuda-bindings` API to do this (requires at least 12.8) command = [f'{CUDA_HOME}/bin/cuobjdump', '-symbols', path] result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) assert result.returncode == 0