[mhk197]

Issue: #9249

Summary

A lambda passed to list_transform can now read values from outside it, such as a column or a parameter of an enclosing lambda. The builder moves each such value out of the body into an extra input of the call and an extra parameter of the lambda. Every element of a row's list sees that row's value.

Stacked on #10503.

list_transform(col("l"), Lambda::new(["x"], checked_add(var("x"), col("a")))?)
// vortex.list.transform($.l, $.a, (x, a) -> (@x + @a))

Changes

• Lambda::extract_captures turns each outside read ($, a path like $.s.a, another lambda's parameter, row_idx(), row_count()) into a new parameter. Lambda::new accepts them; binding still rejects them.
• list_transform(list, lambda) calls it, so the values are ordinary children: replace, field pruning and the scan's row_idx() see them.
• Execution repeats each value once per element of its list with take, using the index of the list that holds each element.
• A bound lambda is rebound instead of erroring when a rebuilt node's inputs change dtype (BoundLambda::rebind).
• With captures the function is not strict: a null captured value keeps its list valid.

Testing

• New tests: a captured column, null and constant captured values, captures in nested lambdas (a column, the outer parameter), rebinding, extract_captures naming, and a scan reading row_idx() inside a lambda (vortex-layout).
• Not run locally yet; CI only.

Follow-up

• Repeat captured values with RunEnd instead of take once RunEnd lives in vortex-array.

[codspeed]

Merging this PR will degrade performance by 6.01%

4 warnings, 1 note

⚠️ Walltime results may be unreliable

Some benchmarks used the Walltime instrument outside CodSpeed Macro Runners, which can make the results inconsistent.

Learn more about Macro Runners

See affected benchmarks

⚠️ Unable to measure the execution of 2 benchmarks

These benchmarks measured no execution time, so their results are not comparable and count as unchanged. This usually happens when the compiler removes the code under test.

Prevent compiler optimizations

See affected benchmarks

⚠️ 24 benchmarks are dominated by system calls

System calls cannot be simulated consistently and are left out of the measure. The reported value understates the real cost. Switch to the Walltime instrument to measure them.

Learn more about system calls

See affected benchmarks

⚠️ Base and head ran on different runner types

Some benchmarks with significant performance changes were compared across different runner types, so the changes might not come from the code. Run base and head on the same runner type, or use CodSpeed Macro Runners for more stable results.

Learn more about Macro Runners

Open the report in CodSpeed to investigate

ℹ️ 40 benchmarks have no flamegraph

The flamegraph wasn't generated: the symbols of the profiling trace could not be resolved, because no information about the language frames was collected.

See affected benchmarks

⚡ 6 improved benchmarks
❌ 12 regressed benchmarks
✅ 2421 untouched benchmarks
⏩ 166 skipped benchmarks¹

Warning

Please fix the performance issues or acknowledge them on CodSpeed.

Performance Changes

	Mode	Benchmark	BASE	HEAD	Efficiency
❌	Walltime	filter_neon[8]	865 ns	1,613 ns	-46.37%
❌	Walltime	threshold_neon[8]	1.4 µs	1.9 µs	-27.52%
❌	Walltime	filter_neon[64]	1.7 µs	2.4 µs	-27.46%
❌	CPU Simulation	encode_primitives[f32, (1000, 2)]	116.4 µs	147.4 µs	-21.01%
❌	Walltime	filter_neon[80]	2.5 µs	3 µs	-17.47%
❌	Walltime	threshold_neon[64]	2.6 µs	3.1 µs	-15.43%
❌	Walltime	threshold_neon[80]	3.4 µs	4.1 µs	-15.13%
❌	CPU Simulation	test_scan_filter_to_arrow_array[int_float]	3.5 ms	4 ms	-13.9%
❌	CPU Simulation	test_repeated_scan_filter[int_float]	3.4 ms	3.8 ms	-11.2%
❌	Walltime	starts_with_avx2[8, VarBinView]	68.5 µs	76.3 µs	-10.21%
❌	CPU Simulation	test_repeated_scan_filter[int]	2.4 ms	2.7 ms	-10.19%
❌	CPU Simulation	test_scan_filter_to_arrow_array[int]	2.5 ms	2.8 ms	-10.17%
⚡	Walltime	mul_u64_nonnull_neon	39.4 µs	20.5 µs	+92.49%
⚡	CPU Simulation	chunked_dict_primitive_canonical_into[f64, (1000, 1000, 10)]	186.7 µs	147.3 µs	+26.77%
⚡	CPU Simulation	missing_stats[16]	84.4 µs	70.6 µs	+19.54%
⚡	Walltime	starts_with_avx2[8, VarBin]	46.6 µs	39.9 µs	+16.8%
⚡	CPU Simulation	take_filter_primitive_small_uncached_random_mask_random_indices[4096, 10]	22.7 µs	19.6 µs	+16.01%
⚡	CPU Simulation	random_i8[0.01]	115.3 µs	101.6 µs	+13.43%

Tip

Investigate this regression by commenting @codspeedbot fix this regression on this PR, or directly use the CodSpeed MCP with your agent.

---

_{Comparing mk/list-transform-captures (c319853) with develop (f62c12f)}

Footnotes

1. 166 benchmarks were skipped, so the baseline results were used instead. If they were deleted from the codebase, click here and archive them to remove them from the performance reports. ↩