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

Apple Silicon/iOS Implementation #5

Open
Proryanator opened this issue Mar 22, 2024 · 2 comments
Open

Apple Silicon/iOS Implementation #5

Proryanator opened this issue Mar 22, 2024 · 2 comments

Comments

@Proryanator
Copy link

Proryanator commented Mar 22, 2024

This is great work! The speed implications are insane.

I'd be interested in making an SHMT framework implementation for apple silicon! Other than this project here for the hardware specified, any active development being done for apple silicon SHMT that you know of?

The non-Nvidia portion should run with docker, but definitely won't run on iOS w/o extra work.

@jk78346
Copy link
Collaborator

jk78346 commented Mar 22, 2024

@Proryanator, thank you for interest. Unfortunately, we don't.
It would be great to have the first Apple silicon implementation of SHMT you are targeting at.

@Proryanator
Copy link
Author

Proryanator commented Mar 22, 2024

@jk78346

https://machinelearning.apple.com/research/neural-engine-transformers

I was reading on how CoreML handles various ML workloads, it sounds analagous to SHMT, especially:

"Core ML then seamlessly blends CPU, GPU, and ANE (if available) to create the most effective hybrid execution plan exploiting all available engines on a given device."

I would need to test this out of course to see what workloads excersize both the GPU/NPU. Reading this doc here it is possible when converting models to CoreML to have it target one specific computing unit, good to know (may be useful for improved optimizations later on).

This graphic here from that article is really nice to be able to see what parts of the system were used when running a CoreML workload:

Screenshot 2024-03-25 at 1 14 42 PM

Will share any findings I make!

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