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

84%

main: 84%

Build Type

push

github

Committed by web-flow

Commit Message

Block hybridization when shape=None would break a rank-dependent body (#838)

An inferred spec whose shape degrades to unknown rank emits
tf.TensorSpec(shape=None, ...), which breaks any body reading its
input's static rank surface: shape.as_list() and len(shape) raise
ValueError, and shape.rank/ndims yield None instead of an integer
(each failure mode runtime-verified on the pinned TF 2.9.3). Teach
StaticShapeReadAnalysis these rank-sensitive sinks as a second read
population resolved by rank alone: they block only when the affected
spec's rank is unknown, tolerating known-rank specs with dynamic axes.
The rank/ndims/len results launder as trace-time integer constants,
which also fixes an over-block: rank arithmetic over a known-rank,
dynamic-axis spec previously tripped the extent-sensitive arithmetic
sink.

Closes #809.


Claude-Session: https://claude.ai/code/session_01GcqABsog1rtb4m2CDZdcRD

Co-authored-by: Claude Fable 5 <noreply@anthropic.com>

Coverage Stats

65 of 69 new or added lines in 2 files covered. (94.2%)

3 existing lines in 1 file now uncovered.

3045 of 3640 relevant lines covered (83.65%)

0.84 hits per line