Skip to content

[codex] classify linear transpose inputs - #45

Draft
shinaoka wants to merge 2 commits into
mainfrom
codex/issue-1321-root-vjp
Draft

[codex] classify linear transpose inputs#45
shinaoka wants to merge 2 commits into
mainfrom
codex/issue-1321-root-vjp

Conversation

@shinaoka

@shinaoka shinaoka commented Jul 7, 2026

Copy link
Copy Markdown
Member

Summary

  • add a classified PrimitiveTransposeInput boundary for transpose rules
  • track linearized graph local values as residual vs tangent-flow values
  • pass residual values and linear values distinctly through linear_transpose
  • update examples/tests for the new transpose-rule input type

Root Cause

linear_transpose normalized every linearized local input into an external value key before invoking primitive transpose rules. That erased whether a value was a residual helper or part of the tangent flow, allowing downstream rules to accidentally retain tangent-only values as runtime tensor operands.

Validation

  • cargo test --workspace

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant