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

[Tracking Issue] Improve CPU backend support without CUDA #187

Open
2 of 3 tasks
LDY1998 opened this issue Apr 23, 2023 · 2 comments
Open
2 of 3 tasks

[Tracking Issue] Improve CPU backend support without CUDA #187

LDY1998 opened this issue Apr 23, 2023 · 2 comments
Labels
enhancement New feature or request

Comments

@LDY1998
Copy link
Contributor

LDY1998 commented Apr 23, 2023

This is a tracking issue for adding CPU compilation without CUDA:

@LDY1998 LDY1998 added the enhancement New feature or request label Apr 23, 2023
@tonyzhao-jt
Copy link

Hi, when will the CPU version be available? Looking forward to it.

@yaoyaoding
Copy link
Member

Hi @SpringWave1, you can try it even now.

a = hidet.randn([3, 4])
b = hidet.randn([4, 3])
c = a @ b

Hidet will generate the matrix multiplication kernel and run it with CPU.
However, now our auto-scheduler only generates kernel that uses a single thread, and there is not efficient matmul schedule template. We are still working on these. The good support might come in the end of this year.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

3 participants