From 9cb9bccbcdc8443b662a706dd0dd07804beb9978 Mon Sep 17 00:00:00 2001
From: fzyzcjy <ch271828n@outlook.com>
Date: Tue, 9 Sep 2025 17:14:06 +0800
Subject: [PATCH] Revert "revert"

This reverts commit a2e1b7e668def1ced3c81a21949d728d7008f428.
---
 flashinfer/jit/core.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/flashinfer/jit/core.py b/flashinfer/jit/core.py
index 3968b2677b..caee57868c 100644
--- a/flashinfer/jit/core.py
+++ b/flashinfer/jit/core.py
@@ -197,6 +197,10 @@ def gen_jit_spec(
         # non debug mode
         cuda_cflags += ["-DNDEBUG"]
 
+    # useful for ncu
+    if bool(os.environ.get("FLASHINFER_JIT_LINEINFO", "0")):
+        cuda_cflags += ["-lineinfo"]
+
     if extra_cflags is not None:
         cflags += extra_cflags
     if extra_cuda_cflags is not None: