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

GPU backend #4

Open
botev opened this issue Jan 23, 2017 · 0 comments
Open

GPU backend #4

botev opened this issue Jan 23, 2017 · 0 comments
Milestone

Comments

@botev
Copy link
Contributor

botev commented Jan 23, 2017

This is a must for when this is to go public so that we can fairly compare against leading frameworks in Machine Learning.

This would require inevitably to be able to generate on the fly new kernels (even based on a template if needed) and being able to compile them. Two options exists if we want to support both Nvidia and AMD devices:

  1. Write separate backends for each of the two vendors (probably focusing Nvidia initially). Things like nvptx and ocl would probably be the starting point of this.

  2. Have a unified backend by using HIP which can allows us to generate same code for any of the two vendors.

Additionally, for comparisons we definitely would need to link cuBLAS and cuDNN for Nvidia devices. It would be nice if we have someone with more expertise in GPU programming who can make suggestions or comments on what is better way of approaching this.

@botev botev added this to the Draft milestone Jan 23, 2017
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

1 participant