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 Support #37

Open
4 tasks
willtebbutt opened this issue Aug 30, 2019 · 1 comment
Open
4 tasks

GPU Support #37

willtebbutt opened this issue Aug 30, 2019 · 1 comment

Comments

@willtebbutt
Copy link
Member

willtebbutt commented Aug 30, 2019

Ensuring that Stheno supports all of its functionality on the GPU (where appropriate) is quite important moving forwards. This will have to be CuArrays.jl-based. To achieve this, we must ensure that:

  • All base mean functions and kernels support GPU inputs
  • All operations on all compositions are implemented in terms of operations that CuArrays can support.
  • logpdf, rand, elbo, etc work correctly with CuArrays.
  • Zygote works efficiently on all of the above.

For the most part, I expect this functionality to require writing quite a number of tests, and making tweaks where necessary.

@willtebbutt
Copy link
Member Author

As regards CI, JuliaGPU have provided these instructions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant