Skip to content

Add test to ensure scan-based and the standard GRU are interchangeable.#8949

Merged
tengyifei merged 1 commit intopytorch:masterfrom
iwknow:gru-interface
Apr 9, 2025
Merged

Add test to ensure scan-based and the standard GRU are interchangeable.#8949
tengyifei merged 1 commit intopytorch:masterfrom
iwknow:gru-interface

Conversation

@iwknow
Copy link
Copy Markdown
Collaborator

@iwknow iwknow commented Apr 7, 2025

this is a preparation step for replacing the standard GRU implementation with the scan-based implementation. for details: #8860

@iwknow
Copy link
Copy Markdown
Collaborator Author

iwknow commented Apr 7, 2025

@tengyifei ensure the scan-based and the standard GRU have the same interface. this is a preparation for replacing the standard with scan-based. please run the tests and review.

@iwknow
Copy link
Copy Markdown
Collaborator Author

iwknow commented Apr 8, 2025

the failed tests are not related to the change. maybe due to the flakiness. please run again or ignore the test result.

@tengyifei tengyifei merged commit d4d997d into pytorch:master Apr 9, 2025
21 of 24 checks passed
@tengyifei
Copy link
Copy Markdown
Collaborator

Thanks!

I think you may ignore GPU failures in the next few days. Still flaky, although we're working on fixing them. This test is covered by TPU and CPU.

@iwknow iwknow deleted the gru-interface branch April 9, 2025 22:01
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.

2 participants