NIFTy-PPL / NIFTy

Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
https://ift.pages.mpcdf.de/nifty/index.html
GNU General Public License v3.0
13 stars 2 forks source link

Gauss-Makrov on GPU #6

Closed phaim closed 4 months ago

phaim commented 4 months ago

The the loop in the general gm-process introduced in this commit leads to significant performance degradation of the correlated field on the GPU. This seems to be caused by this loop. A potential fix could be to implement a specialized function where it is changed to a cumsum and use that function in the correlated field.

Edenhofer commented 4 months ago

@ph-frank Can you take care of this?

ph-frank commented 4 months ago

Fixed. See Merge request ift/nifty!927