Reland upstream commit 0ba4205 (#3320)

Author: anmyachev

python/test/unit/language/test_line_info.py

@@ -250,3 +250,24 @@ def test_line_info_interpreter(func: str):
         expected_def_lineno = 68
     kernel.rewrite()
     assert kernel.rewriter.def_file_lineno == expected_def_lineno
+
+
+@pytest.mark.parametrize("status", ["0", "1"])
+def test_line_info_env(monkeypatch, status: str):
+    if is_interpreter():
+        pytest.skip("interpreter does not support warmup compilation")
+
+    try:
+        obj_kind, command, anchor, separator = get_disassembler_command_and_debug_line_format()
+    except BaseException:
+        pytest.skip("disassembler is not available")
+
+    shape = (128, )
+    monkeypatch.setenv("TRITON_DISABLE_LINE_INFO", status)
+    kernel_single.device_caches.clear()
+    kernel_info = kernel_single.warmup(torch.float32, torch.float32, BLOCK=shape[0], grid=(1, ))
+    if obj_kind == "spvbin":
+        file_lines = spv_extract_file_lines(kernel_info.asm["spv"], command)
+    else:
+        file_lines = extract_file_lines(command, anchor, separator, kernel_info.asm[obj_kind])
+    assert len(file_lines) == 0 if status == "1" else len(file_lines) > 0

third_party/nvidia/backend/compiler.py

@@ -393,7 +393,8 @@ def make_cubin(self, src, metadata, opt, capability):
             fsrc.flush()
             fbin = fsrc.name + '.o'
 
-            line_info = [] if os.environ.get('TRITON_DISABLE_LINE_INFO') else ['-lineinfo']
+            line_info = ["-lineinfo", "-suppress-debug-info"] if os.environ.get("TRITON_DISABLE_LINE_INFO",
+                                                                                "0") == "1" else ["-lineinfo"]
             fmad = [] if opt.enable_fp_fusion else ['--fmad=false']
             suffix = 'a' if capability >= 90 else ''
             opt_level = ['--opt-level', '0'] if os.environ.get("DISABLE_PTXAS_OPT", "0") == "1" else []