Skip to content

[stale] Clean feature history#67

Open
stevenvar wants to merge 16 commits intojoeyye-work:new-serving-2.20from
stevenvar:steven.joey_for_serving_history_by_feature_clean
Open

[stale] Clean feature history#67
stevenvar wants to merge 16 commits intojoeyye-work:new-serving-2.20from
stevenvar:steven.joey_for_serving_history_by_feature_clean

Conversation

@stevenvar
Copy link
Copy Markdown
Collaborator

No description provided.

joeyye-work and others added 16 commits December 3, 2025 20:43
Adding --tf_xla_annotate_cluster_id to allow operator name
starting with .cluster.id to influcence clustering decisions
Adds the CPU runtime plumbing needed to carry the dynamic batch value through execution and lowers the supporting runtime pieces.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Adds batch matching support and updates output handling so compiled executions can normalize and restore batch-aware values.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Introduces symbolic dimension expressions to TensorFlow and XLA shape representations so dynamic dimensions can be tracked structurally instead of only by concrete values.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Captures dynamic dimension values at compile time and adds clustering safeguards so incompatible dynamic-shape regions are not merged accidentally.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Keeps inferred shape metadata alive when values cross JIT and encapsulation boundaries so downstream compilation can still recover symbolic shape information.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Tightens clustering and shape-inference behavior around partially known or unranked shapes to avoid losing dynamic-shape information in unsupported cases.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Improves tf2xla shape propagation and normalizes padded constants so shape-derived values can continue through lowering in a consistent way.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Fixes the reshape ratio handling so shape-expression information stays consistent when reshape dimensions are derived from symbolic values.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Limits the Eigen fallback path to dynamic CPU dot cases so static cases continue to use the normal lowering while dynamic dimensions remain supported.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Shares common expression-inference helpers across passes so symbolic dimension analysis stays aligned in different parts of the pipeline.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
…operators

Extends expression updates across additional TensorFlow operators so symbolic dimensions continue to reflect the same shape transformations as concrete dimensions.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
…mic_sizes

Removes the separate single-dynamic-dimension flag and reuses the existing dynamic-sizes flag so the feature is configured in one place.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Cleans up remaining symbolic-expression propagation details and aligns edge cases discovered during integration of the earlier feature commits.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
Fixes shape-expression bookkeeping after removing dimensions so the expression vector stays aligned with the resulting shape.

Co-authored-by: Steven Varoumas <steven.varoumas1@huawei.com>
Co-authored-by: Utku Saglam <utku.saglam@h-partners.com>
Co-authored-by: Jinyun (Joey) Ye <jinyunye@huawei.com>
Co-authored-by: Muteages <67578152+Muteages@users.noreply.github.com>
Co-authored-by: Guillermo Callaghan <guillermo2@huawei.com>
@stevenvar stevenvar force-pushed the steven.joey_for_serving_history_by_feature_clean branch 3 times, most recently from 4b15ce8 to 3eb8a9e Compare March 21, 2026 23:05
@stevenvar stevenvar added the enhancement New feature or request label Mar 21, 2026
@stevenvar stevenvar requested a review from joeyye-work March 21, 2026 23:17
@stevenvar stevenvar added the invalid This doesn't seem right label Mar 31, 2026
@stevenvar stevenvar changed the title Clean feature history [stale] Clean feature history Mar 31, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request invalid This doesn't seem right

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants