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

MPS device compatibility #5

Open
adills opened this issue Sep 15, 2023 · 2 comments
Open

MPS device compatibility #5

adills opened this issue Sep 15, 2023 · 2 comments

Comments

@adills
Copy link

adills commented Sep 15, 2023

With Pytorch and TF both supporting Apple's M1/M2 silicon chips, you should be able to include a device checker and include "mps" capability for GPU processing. If only I had the time to create a branch and show you.

@Spycsh
Copy link
Owner

Spycsh commented Sep 15, 2023

Thanks for info:) I do not have access to a Mac currently and the optimization here in my repo should only be used for intel xeon cpu. But good to know about the mps device compatibility. Is it fast enough?

@adills
Copy link
Author

adills commented Sep 15, 2023

Thank you for your quick reply. I can run inference.py using your int8 instructions and --cpu on my MacBook M1 Max Pro but at a rate of ~18s/it (not even 1 it/s). If I get some time, I might create a fork and see if I can modify your engine to accept device=mps.

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

No branches or pull requests

2 participants