Skip to content

Added option to bypass getitem traces in stateful modules#49

Merged
benkroehs merged 1 commit into
mainfrom
state-assignment
Jan 23, 2026
Merged

Added option to bypass getitem traces in stateful modules#49
benkroehs merged 1 commit into
mainfrom
state-assignment

Conversation

@Jegp

@Jegp Jegp commented Jan 18, 2026

Copy link
Copy Markdown
Collaborator
  • Fixed tracing for stateful modules that return a tuple of (values, state) by ignoring the state nodes
  • Added options to list stateful modules when tracing them, so libraries to explicitly inform the tracing when to ignore state
  • Added concrete_args for tracing hints

@Jegp Jegp requested a review from benkroehs January 18, 2026 21:07
benkroehs
benkroehs approved these changes Jan 18, 2026
View reviewed changes

@benkroehs benkroehs left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me :)

@benkroehs benkroehs merged commit 18d32a2 into main Jan 23, 2026
15 checks passed
@Jegp Jegp deleted the state-assignment branch January 23, 2026 18:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@benkroehs benkroehs benkroehs approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@Jegp @benkroehs