Skip to content

Use int4mm weight packing MPS kernel #866

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
Jun 26, 2024
Merged

Use int4mm weight packing MPS kernel #866

merged 2 commits into from
Jun 26, 2024

Conversation

manuelcandales
Copy link
Contributor

This should land after the following PyTorch PR lands: pytorch/pytorch#128965
in order to use the int4mm weight packing MPS kernel introduced there

Copy link

pytorch-bot bot commented Jun 18, 2024

🔗 Helpful Links

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

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

✅ No Failures

As of commit 0131b58 with merge base d71783c (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 Jun 18, 2024
@Jack-Khuu Jack-Khuu self-requested a review June 18, 2024 21:22
@Jack-Khuu
Copy link
Contributor

Heads up that I have some PR's that'll land to fix most of these test (CI is just broke broke atm)

Copy link
Contributor

@malfet malfet left a comment

Choose a reason for hiding this comment

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

LGTM, but you'll need to update pytorch pin (otherwise your changes would not be visible)

@manuelcandales manuelcandales merged commit 81fce9c into main Jun 26, 2024
51 checks passed
@jerryzh168
Copy link
Contributor

so int4mm path for mps is enabled now?

malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly

Updating torch nightly to pick up int4mm groupsize 256 fix (#874)
malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly
malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly
malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly
malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly

Updating torch nightly to pick up int4mm groupsize 256 fix (#874)
malfet pushed a commit that referenced this pull request Jul 17, 2024
* Use int4mm weight packing mps kernel

* update torch nightly

Updating torch nightly to pick up int4mm groupsize 256 fix (#874)
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.

5 participants