TST: Test with multi-GPU on CI

AdvancedPhotonSource / tike

Repository for ptychography software

http://tike.readthedocs.io

Other

29 stars 15 forks source link

Closed carterbox closed 3 years ago

carterbox commented 3 years ago

The purpose of this PR is to start testing multi-GPU code paths on our CI.

carterbox commented 3 years ago

This doesn't work because you cannot make views of GPU arrays outside of their device's context. i.e.

with cp.cuda.Device(0):
    y = x[:, indices] # not allowed if x is located on device 1

carterbox commented 3 years ago

This problem with be fixed in #104.