Skip to content

Handle random seed for eager mode#7328

Merged
JackCaoG merged 2 commits intomasterfrom
JackCaoG/eager_random_seed
Jun 25, 2024
Merged

Handle random seed for eager mode#7328
JackCaoG merged 2 commits intomasterfrom
JackCaoG/eager_random_seed

Conversation

@JackCaoG
Copy link
Copy Markdown
Collaborator

seed_ir_value was reset by mark_step but we don't call mark_step in eager mode. In eager mode it make more sense to manually rotate the devctx->seed and always create a new seed IR

@JackCaoG JackCaoG added the eager PyTorch/XLA eager-mode label Jun 21, 2024
@JackCaoG JackCaoG requested review from alanwaketan and lsy323 June 21, 2024 23:30
@JackCaoG
Copy link
Copy Markdown
Collaborator Author

ok this pr is ready for review

@JackCaoG JackCaoG requested a review from wonjoo-wj June 25, 2024 00:23
@JackCaoG JackCaoG merged commit f570830 into master Jun 25, 2024
JackCaoG added a commit that referenced this pull request Jul 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

eager PyTorch/XLA eager-mode

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants