`altar.cuda.distributions.cudaDistribution`

Module Contents

Classes

class altar.cuda.distributions.cudaDistribution.cudaDistribution(name, locator, **kwds)

Bases: altar.distributions.Base.Base

The base class for probability distributions

parameters

doc = the number of model parameters that belong to me

offset

doc = the starting point of my parameters in the overall model state

device

idx_range

precision

initialize(self, rng): Initialize with the given random number generator

verify(self, theta, mask): Check whether my portion of the samples in {theta} are consistent with my constraints, and update {mask}, a vector with zeroes for valid samples and non-zero for invalid ones

cuInitialize(self, application): cuda specific initialization

cuInitSample(self, theta): cuda process to initialize random samples

cuVerify(self, theta, mask): cuda process to verify the validity of samples

cuEvalPrior(self, theta, prior): cuda process to compute the prior