ponder-lab / Hybridize-Functions-Refactoring / #2384

81%

main: 83%

Build Type

Pull #748

github

Committed by khatchad

Commit Message

Make the parameter-flow numpy precondition shape-aware (two-color taint)

The conservative call-site taint (#746) declined any function reaching numpy through a parameter, including functions that apply numpy only to a tensor's *shape* — a false positive for the common `get_shape_list`/`einsum_via_matmul` idiom, whose numpy operates on static shape dimensions and does not crash.

The scan now tracks two taint colors. A *value* taint marks the tensor value itself, over which numpy always raises under tracing; a *shape* taint marks a value derived from a tensor's shape (`.shape`/`.shape.as_list()`, `tf.shape`/`size`/`rank`), over which numpy is graph-compatible for static shapes. numpy over a value-tainted argument is a sink (decline); over a shape-tainted argument it is permitted. Call sites are tainted conservatively (following comprehension element-through-container flow, so `TUDataset.cat` is still caught), but the result is colored SHAPE when the callee is a pure shape extractor — its value taint is consumed solely by shape operations and never escapes (`get_shape_list`) — recovering `DenseLayer3d.call`/`DenseLayer3dProj.call` as hybridizable.

This is precision-favoring on the shape axis: numpy over a shape-derived value is treated as safe, sound when the consumed dimensions are static (as across the corpus). Replaces the earlier shape-staticness sanitizer, which was corpus-neutral and could not recover the interprocedural case. `testNumpyOnShapeThroughHelper` pins the recovery; the five true crashers and `cat` still decline (full suite green, 565).

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01He1ULx8nRoHzGicMryry1A

Pull Request Pull Request #748: Make the parameter-flow numpy precondition shape-aware

Coverage Stats

83 of 84 new or added lines in 1 file covered. (98.81%)

14 existing lines in 1 file now uncovered.

1976 of 2437 relevant lines covered (81.08%)

0.81 hits per line