-
Notifications
You must be signed in to change notification settings - Fork 106
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
WIP: GPU tensors #739
WIP: GPU tensors #739
Conversation
odl/space/gpuary_ntuples.py
Outdated
lico(a, x1.data, b, x2.data, out.data) | ||
|
||
|
||
class GpuArrayTensorSpace(FnBase, GpuArrayRawTensorSpace): |
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.
From here on, the rest is currently copy-paste from npy_ntuples.py
. Ignore it.
b10ab58
to
2f1cb4b
Compare
994c6a1
to
8e8ba6d
Compare
24b4ea9
to
5bd5b45
Compare
fff61da
to
4ca2ddc
Compare
49cb419
to
330cbb4
Compare
362b26c
to
3373171
Compare
Great news:
|
I know - but good to hear that they are trying to make it work for Windows, too. |
19dc7bf
to
d8701ec
Compare
c25c793
to
9988b49
Compare
076fccc
to
97bba94
Compare
b2185c3
to
2eb7662
Compare
Second WIP PR.
Much earlier stage than #738 but maybe some comments already applicable.TODOs:
pygpu
They should.No, becausereshape
ofGpuArray
objects copy the array, in contrast to Numpy.pygpu
is not installed. Should the package be mandatory?Update: manual doctests are fixed by adding a conditional in the
__main__
section, need to fix runs bypytest
also.gpuary_tensors.py
namespace clean by making context initialization code a function only returning the relevant stuff.