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

CPU Port #7

Open
okpatil4u opened this issue Jan 25, 2023 · 4 comments
Open

CPU Port #7

okpatil4u opened this issue Jan 25, 2023 · 4 comments

Comments

@okpatil4u
Copy link

How can I port these models to a CPU only code ? I have 128 GB Mac M1 Ultra Studio system which has 21 Tflops throughput, which render the use of block fftconv moot.

Is it possible to create a CPU only model ? How should one start working on it ?

@DanFu09
Copy link
Contributor

DanFu09 commented Jan 25, 2023

This is on our roadmap - there's some dependencies that we need to work through for the SSM support (just kernel generation, not even the block FFTConv yet).

Thank you for posting the issue! It's helpful to get a sense of the platforms that people want to run on.

@okpatil4u
Copy link
Author

Thanks Dan. After the changes in your roadmap, can we benchmark the huggingface weights on CPU then ?

@DanFu09
Copy link
Contributor

DanFu09 commented Jan 25, 2023 via email

@okpatil4u
Copy link
Author

Hey Dan, do you have any timelines in mind for this ?

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