Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Check size on transformer cache #94

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

graemenail
Copy link
Member

Description

Copy of marian-nmt#942 to fix segfaults as first described in PR marian-nmt#881.

List of changes:

  • Check shape instead of elements in transformer cache retrieval

Added dependencies: none

How to test

Passes CI
To be tested with model exhibiting the issue.

Checklist

  • I have tested the code manually
  • I have run regression tests
  • I have read and followed CONTRIBUTING.md
  • I have updated CHANGELOG.md

@graemenail graemenail force-pushed the bmt-fix-transformer-cache branch from ca0aeb2 to 33fd8d7 Compare August 24, 2022 13:40
@XapaJIaMnu
Copy link
Collaborator

Is this the final version of the fixes?

@graemenail graemenail marked this pull request as ready for review September 27, 2022 18:57
@graemenail
Copy link
Member Author

Yes @XapaJIaMnu, feel free to merge whenever

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