From 1b9b4a0fd5da8827587645ada6bad6051788f101 Mon Sep 17 00:00:00 2001 From: haok1402 Date: Wed, 31 Dec 2025 16:21:44 -0500 Subject: [PATCH] add a `kernel_global_source` property to `TVMFFIKernelAdapter` that returns `device_kernel_source` --- tilelang/jit/adapter/tvm_ffi.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tilelang/jit/adapter/tvm_ffi.py b/tilelang/jit/adapter/tvm_ffi.py index cd473efad..47f873ec8 100644 --- a/tilelang/jit/adapter/tvm_ffi.py +++ b/tilelang/jit/adapter/tvm_ffi.py @@ -97,6 +97,7 @@ def __init__( self.pass_configs = pass_configs self.compile_flags = compile_flags self.dynamic_symbolic_map = self._process_dynamic_symbolic() + self.kernel_global_source = self.device_kernel_source self._post_init()