Fix bug in output sampler on GPU

awslabs / fortuna

A Library for Uncertainty Quantification.

https://aws-fortuna.readthedocs.io/en/latest/

Apache License 2.0

892 stars 46 forks source link

Closed gianlucadetommaso closed 11 months ago

gianlucadetommaso commented 11 months ago

There is a small bug in the output calibrator sampler due to sharding, which manifests only on CPU. This PR fixes it.