Skip to content

[Distributed] Fix cache lane #1194

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

Merged
merged 2 commits into from
Sep 24, 2024
Merged

[Distributed] Fix cache lane #1194

merged 2 commits into from
Sep 24, 2024

Conversation

kwen2501
Copy link
Contributor

@kwen2501 kwen2501 commented Sep 24, 2024

Previous PR #1174 has two issues:

(i) Wrong default cache_lane value given to forward function (should be 0, gave 1).

(ii) Did not add cache lane impl to CustomKVCache. The impl is required if we want the KVCache to CustomKVCache replacement work smoothly.

Copy link

pytorch-bot bot commented Sep 24, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchchat/1194

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 3166c62 with merge base 04ea309 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Sep 24, 2024
@kwen2501 kwen2501 requested a review from lessw2020 September 24, 2024 18:26
Copy link
Contributor

@lessw2020 lessw2020 left a comment

Choose a reason for hiding this comment

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

thanks for the update - lgtm.

@kwen2501 kwen2501 merged commit 6d2ef4a into main Sep 24, 2024
51 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants