ponder-lab / Hybridize-Functions-Refactoring / 27304054625

79%

push

github

Implement the `RECONFIGURE` transformation to emit `input_signature` (#595)

* Implement the `RECONFIGURE` transformation to emit `input_signature`

Close the loop between input-signature inference and the source rewrite for
already-hybrid functions: when inference is enabled and a qualifying hybrid
`@tf.function` carries no `input_signature`, add the inferred
`input_signature=[tf.TensorSpec(...)]` to the existing decorator.

- Add `PreconditionSuccess.P4` for the reconfigure case.
- Select `RECONFIGURE` + `P4` in `Function.check()`'s no-primitive-parameters
  hybrid branch, gated on the `inferInputSignatures` flag (default off, so the
  precondition matrix is unchanged for existing tests), no existing
  `input_signature` (the supplied-signature case is deferred per the #557
  contract), no Python side-effects, non-recursive, and an inferrable signature.
- Implement `Function.reconfigure()`, reusing the import-shape resolution and
  emission helpers (`getImportContext`, `computeInputSignatureKeyword`,
  `addInputSignature`). Handles the bare `@tf.function`, empty-parentheses
  `@tf.function()`, and non-empty `@tf.function(...)` decorator forms.
- Dispatch `RECONFIGURE` in `transform()` (was `UnsupportedOperationException`).
- Add reconfigure tests and fixtures, including the deferred-supplied-signature
  and flag-off precondition-matrix-unchanged guards.

The eager `CONVERT_TO_HYBRID` emission half landed in #565; this completes the
already-hybrid half.

Closes #563

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* Document `P4` as the add-when-absent path into `RECONFIGURE`

Note that modifying an already-supplied `input_signature` (validate-then-overwrite)
is a separate precondition path reaching the same `RECONFIGURE` transformation,
mirroring how `P2` and `P3` both reach `CONVERT_TO_EAGER`.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* Gate `RECONFIGURE` selection on input-signature emittability

Only select `RECONFIGURE`/`... (continued)

50 of 52 new or added lines in 2 files covered. (96.15%)

1443 of 1827 relevant lines covered (78.98%)

0.79 hits per line