https://github.com/jofrn created https://github.com/llvm/llvm-project/pull/200351
[lit] Add lit.llvm.fn_extract: --param fn=NAMES prepends llvm-extract >From f6ab0a77626aa00649329e623b8eb3df589f2e09 Mon Sep 17 00:00:00 2001 From: jofrn <[email protected]> Date: Fri, 29 May 2026 01:12:01 -0700 Subject: [PATCH] [lit] Add lit.llvm.fn_extract: --param fn=NAMES prepends llvm-extract --- llvm/utils/lit/lit/llvm/fn_extract.py | 19 ++++++++++++++ llvm/utils/lit/lit/llvm/fn_param.py | 6 ++--- .../utils/lit/tests/Inputs/fn-extract/lit.cfg | 19 ++++++++++++++ .../Inputs/fn-extract/mock-bin/llvm-extract | 2 ++ .../lit/tests/Inputs/fn-extract/sample.ll | 2 ++ llvm/utils/lit/tests/fn-extract.py | 26 +++++++++++++++++++ 6 files changed, 71 insertions(+), 3 deletions(-) create mode 100644 llvm/utils/lit/lit/llvm/fn_extract.py create mode 100644 llvm/utils/lit/tests/Inputs/fn-extract/lit.cfg create mode 100755 llvm/utils/lit/tests/Inputs/fn-extract/mock-bin/llvm-extract create mode 100644 llvm/utils/lit/tests/Inputs/fn-extract/sample.ll create mode 100644 llvm/utils/lit/tests/fn-extract.py diff --git a/llvm/utils/lit/lit/llvm/fn_extract.py b/llvm/utils/lit/lit/llvm/fn_extract.py new file mode 100644 index 0000000000000..f7d7f1d236a07 --- /dev/null +++ b/llvm/utils/lit/lit/llvm/fn_extract.py @@ -0,0 +1,19 @@ +"""Prepend `llvm-extract --func=NAME %s -o -` to the first pipeline stage of +every RUN line that reads from `%s`, so subsequent tools (opt/llc/llvm-as/...) +operate on a module narrowed to the named functions and their transitive +dependencies. Tool-agnostic. Driven by `--param fn=NAMES`.""" + +from lit.llvm.fn_param import add_capture_sub, parse_fn_names + + +def install(config, lit_config): + names = parse_fn_names(lit_config) + if not names: + return + extract = "llvm-extract " + " ".join("--func=" + n for n in names) + " %s -o -" + # Match: optional %dbg(...) marker, then the first pipeline stage up to + # (but not crossing) the first `|`, ending with `< %s` or bare `%s`. + # `\s*` immediately before `%s` is consumed so we don't leave a double + # space after stripping the redirect/positional. + pattern = r"^(%dbg\([^)]*\)\s*)?([^|]*?)\s*(?:<\s*)?%s(?=\s|$)" + add_capture_sub(config, pattern, r"\1" + extract + r" | \2") diff --git a/llvm/utils/lit/lit/llvm/fn_param.py b/llvm/utils/lit/lit/llvm/fn_param.py index 616f519c32c95..ebab824674dab 100644 --- a/llvm/utils/lit/lit/llvm/fn_param.py +++ b/llvm/utils/lit/lit/llvm/fn_param.py @@ -32,6 +32,6 @@ def install(config, lit_config): # fn_selection.install(config, lit_config) pass else: - # from lit.llvm import fn_extract - # fn_extract.install(config, lit_config) - pass + from lit.llvm import fn_extract + + fn_extract.install(config, lit_config) diff --git a/llvm/utils/lit/tests/Inputs/fn-extract/lit.cfg b/llvm/utils/lit/tests/Inputs/fn-extract/lit.cfg new file mode 100644 index 0000000000000..4452e2f71829c --- /dev/null +++ b/llvm/utils/lit/tests/Inputs/fn-extract/lit.cfg @@ -0,0 +1,19 @@ +import lit.formats +from lit.llvm import fn_param + +config.name = "fn-extract" +config.suffixes = [".ll"] +config.test_format = lit.formats.ShTest() +config.test_source_root = None +config.test_exec_root = None + +# Make a fake `llvm-extract` (exits 0) discoverable, so the prepended pipeline +# stage succeeds and lit's -a output shows the substituted command line for +# FileCheck to inspect. +config.environment["PATH"] = ( + os.path.join(os.path.dirname(__file__), "mock-bin") + + os.pathsep + + config.environment.get("PATH", "") +) + +fn_param.install(config, lit_config) diff --git a/llvm/utils/lit/tests/Inputs/fn-extract/mock-bin/llvm-extract b/llvm/utils/lit/tests/Inputs/fn-extract/mock-bin/llvm-extract new file mode 100755 index 0000000000000..039e4d0069c5c --- /dev/null +++ b/llvm/utils/lit/tests/Inputs/fn-extract/mock-bin/llvm-extract @@ -0,0 +1,2 @@ +#!/bin/sh +exit 0 diff --git a/llvm/utils/lit/tests/Inputs/fn-extract/sample.ll b/llvm/utils/lit/tests/Inputs/fn-extract/sample.ll new file mode 100644 index 0000000000000..d1c6ec435d53e --- /dev/null +++ b/llvm/utils/lit/tests/Inputs/fn-extract/sample.ll @@ -0,0 +1,2 @@ +; RUN: echo opt %s -S -passes='instcombine' | echo FileCheck %s +; RUN: echo opt < %s -S -passes="instcombine" | echo FileCheck %s diff --git a/llvm/utils/lit/tests/fn-extract.py b/llvm/utils/lit/tests/fn-extract.py new file mode 100644 index 0000000000000..7411a44edacb5 --- /dev/null +++ b/llvm/utils/lit/tests/fn-extract.py @@ -0,0 +1,26 @@ +# Verify --param fn=NAMES prepends llvm-extract to the first pipeline stage +# via lit.llvm.fn_extract (which is also wired into llvm/test/lit.cfg.py). + +# --- --param fn=foo: single function --- +# RUN: %{lit} -a --param fn=foo %{inputs}/fn-extract/sample.ll \ +# RUN: | FileCheck --check-prefix=SINGLE %s +# +# Positional `%s` form: +# SINGLE: llvm-extract --func=foo {{.*}}sample.ll -o - | echo opt -S -passes='instcombine' | echo FileCheck {{.*}}sample.ll +# Redirect `< %s` form: +# SINGLE: llvm-extract --func=foo {{.*}}sample.ll -o - | echo opt -S -passes="instcombine" | echo FileCheck {{.*}}sample.ll + +# --- --param fn=foo,bar: multiple functions --- +# RUN: %{lit} -a --param fn=foo,bar %{inputs}/fn-extract/sample.ll \ +# RUN: | FileCheck --check-prefix=MULTI %s +# +# MULTI: llvm-extract --func=foo --func=bar {{.*}}sample.ll -o - | echo opt -S -passes='instcombine' | echo FileCheck {{.*}}sample.ll +# MULTI: llvm-extract --func=foo --func=bar {{.*}}sample.ll -o - | echo opt -S -passes="instcombine" | echo FileCheck {{.*}}sample.ll + +# --- No --param: pipeline unchanged --- +# RUN: %{lit} -a %{inputs}/fn-extract/sample.ll \ +# RUN: | FileCheck --check-prefix=NONE %s +# +# NONE-NOT: llvm-extract +# NONE: echo opt {{.*}}sample.ll -S -passes='instcombine' +# NONE: echo opt < {{.*}}sample.ll -S -passes="instcombine" _______________________________________________ llvm-branch-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
