search

bd-j / forcepho

Generative modeling galaxy photometry for JWST
https://forcepho.readthedocs.io
19 stars 4 forks source link

Port to other GPUs #33

Closed bd-j closed 4 years ago

bd-j commented 5 years ago

It will be useful to have the GPU code able to run on non-summit GPUs to which we have access.

One Volta specific feature that may not be easy to replicate is the MPS server that partitions the GPU in space instead of time. If it is true that pre-Volta MPS cannot efficiently use the GPU from multiple CPU processes, we can try to emulate MPS using a single CPU MPI rank that mediates between the sampler processes and the GPU. We can also simply live with some inefficiency until a time closer to launch when it makes sense to obtain access to a Volta GPU.

bd-j commented 4 years ago

V100s are readily available; porting does not seem to be required (though a CPU port might be nice). Closing this.