Home
last modified time | relevance | path

Searched refs:sampler (Results 1 – 4 of 4) sorted by relevance

/petsc/src/dm/dt/interface/
H A Ddtprob.c531 …onst char prefix[], const char name[], PetscProbFn **pdf, PetscProbFn **cdf, PetscProbFn **sampler) in PetscProbCreateFromOptions() argument
548 if (sampler) { in PetscProbCreateFromOptions()
549 PetscAssertPointer(sampler, 6); in PetscProbCreateFromOptions()
550 *sampler = NULL; in PetscProbCreateFromOptions()
558 if (sampler) *sampler = PetscPDFSampleConstant1D; in PetscProbCreateFromOptions()
563 if (sampler) *sampler = PetscPDFSampleConstant2D; in PetscProbCreateFromOptions()
568 if (sampler) *sampler = PetscPDFSampleConstant3D; in PetscProbCreateFromOptions()
579 if (sampler) *sampler = PetscPDFSampleGaussian1D; in PetscProbCreateFromOptions()
583 if (sampler) *sampler = PetscPDFSampleGaussian2D; in PetscProbCreateFromOptions()
587 if (sampler) *sampler = PetscPDFSampleGaussian3D; in PetscProbCreateFromOptions()
/petsc/src/dm/dt/tests/
H A Dex14.c44 PetscProbFn *sampler[2] = {PetscPDFSampleGaussian1D, PetscPDFSampleGaussian2D}; in TestSampling() local
65 PetscCall(sampler[s](r, NULL, o)); in TestSampling()
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opus.cu168 PetscMatrixSampler *sampler; member
222 delete a->sampler; in MatDestroy_H2OPUS()
732 if (!c && a->sampler) { in MatH2OpusInferCoordinates_Private()
733 Mat S = a->sampler->GetSamplingMat(); in MatH2OpusInferCoordinates_Private()
912 if (!kernel && a->sampler && a->sampler->GetSamplingMat()) { in MatAssemblyEnd_H2OPUS()
918 …PetscCall(MatApproximateNorm_Private(a->sampler->GetSamplingMat(), NORM_2, a->norm_max_samples, &A… in MatAssemblyEnd_H2OPUS()
920 …if (a->sf && !a->nativemult) a->sampler->SetIndexMap(a->hmatrix->u_basis_tree.index_map.size(), a-… in MatAssemblyEnd_H2OPUS()
921 a->sampler->SetStream(handle->getMainStream()); in MatAssemblyEnd_H2OPUS()
923 a->sampler->SetGPUSampling(false); in MatAssemblyEnd_H2OPUS()
924 …hara(a->sampler, *a->hmatrix, a->max_rank, 10 /* TODO */, a->rtol * Anorm, a->bs, handle, verbose); in MatAssemblyEnd_H2OPUS()
[all …]
/petsc/src/dm/impls/swarm/
H A Dswarmpic.c1213 PetscErrorCode DMSwarmInitializeVelocities(DM sw, PetscProbFn *sampler, const PetscReal v0[]) in DMSwarmInitializeVelocities() argument
1251 PetscCall(sampler(a, NULL, vel)); in DMSwarmInitializeVelocities()
1276 PetscProbFn *sampler; in DMSwarmInitializeVelocitiesFromOptions() local
1295 …cCall(PetscProbCreateFromOptions(dim, prefix, "-dm_swarm_velocity_density", NULL, NULL, &sampler)); in DMSwarmInitializeVelocitiesFromOptions()
1296 PetscCall(DMSwarmInitializeVelocities(sw, sampler, v0)); in DMSwarmInitializeVelocitiesFromOptions()