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

90%

push

github

Leave Python scalars and len out of the shape-value boundary walk (#1228)

* Leave Python scalars and len out of the shape-value boundary walk

A Python scalar argument keeps its value under tf.function, since it is
part of the trace key, and len reads a tensor's static length into a
Python int; neither loses a shape value at the boundary, so neither
should make PASSES_SHAPE_VALUE_ACROSS_BOUNDARY refuse (#1225).

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01NTnpjPjJj2aZy1ZkUD1eAn

* Keep a parameter an unmodeled argument reaches as a shape-value source

A call passing a value with no abstract object adds nothing to the
parameter's points-to set, so a parameter it reaches could look as
though only Python scalars reach it. Also restate why len launders:
traced, it is a constant or tf.shape(x)[0], both read from the type.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01NTnpjPjJj2aZy1ZkUD1eAn

* Follow an argument through merges before trusting its points-to set

A value with no abstract object disappears once a phi merges it with a
constant, or once a caller passes it on through its own parameter, so a
parameter it reaches could still look as though only Python scalars reach
it. The check now follows phi operands and the caller's own parameters
back to their call sites. Also pin that an index operand keeps its color
inbound.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01NTnpjPjJj2aZy1ZkUD1eAn

* Follow a call's result to its callee's returns

A callee's returns merge as a phi's operands do, so a call returning
either a constant or a value with no abstract object has a result whose
points-to set holds only the constant. Also say that a value read from an
attribute, like one read from a container, isn't followed.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Sessi... (continued)

74 of 83 new or added lines in 3 files covered. (89.16%)

6115 of 6829 relevant lines covered (89.54%)

0.9 hits per line