Compatibility with `cupy`
Since gwpopulation
runs most efficiently on a GPU, this code base gw-distributions
needs to be compatible with cupy
in order to be fully optimized in its compatibility with gwpopulation
. This is implemented already through backends
(see !1 (merged)), but has not been thoroughly tested and troubleshooted (see !15 ).
In order to implement this, the following benchmarks need to be completed:
-
draw events from one of each type of SamplingDistribution
while settingbe.use('cupy')
(one of each meaning one redshift, one mass, etc) -
draw events from an EventGenerator
that has multipleSamplingDistribution
s while setting settingbe.use('cupy')
-
run gwpopulation
withgw-distribution
on a GPU
Edited by Amanda Farah