-
Notifications
You must be signed in to change notification settings - Fork 2
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
Basic GPU Support #118
Basic GPU Support #118
Conversation
…eature/gpu-support
…works.jl into feature/gpu-support
…works.jl into feature/gpu-support
Main thing to change is removing scalar indexing when constructing parameterisations. Scalar indexing is not done on the GPU which can lead to very expensive data transfer back and forth between CPU and GPU. |
There's a scalar indexing warning popping up on the backwards passes of |
This seemed like a good time to change the |
Feature/gpu support debug
There was a rather frustrating bug when evaluating the RENs on a GPU. In the This was causing the REN to have different outputs when evaluated multiple times with the same inputs. Sometimes, the outputs would be |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good!
Adding basic GPU support in accordance with #105.