CFD/hypre
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'nvcollab' of github.com:hypre-space/hypre into nvcollab

Browse Source
This commit is contained in:
Ruipeng Li 2022-03-11 08:34:20 -08:00
commit 793b22aaf0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/utilities/device_utils.c
View File

@ -1450,11 +1450,12 @@ HYPRE_Int
hypre_ResetDeviceRandGenerator( hypre_ulonglongint seed,
hypre_ulonglongint offset )
{
curandGenerator_t gen = hypre_HandleCurandGenerator(hypre_handle());
#if defined(HYPRE_USING_CURAND)
curandGenerator_t gen = hypre_HandleCurandGenerator(hypre_handle());
HYPRE_CURAND_CALL( curandSetPseudoRandomGeneratorSeed(gen, seed) );
HYPRE_CURAND_CALL( curandSetGeneratorOffset(gen, offset) );
#elif defined(HYPRE_USING_ROCRAND)
rocrand_generator gen = hypre_HandleCurandGenerator(hypre_handle());
HYPRE_ROCRAND_CALL( rocrand_set_seed(gen, seed) );
HYPRE_ROCRAND_CALL( rocrand_set_offset(gen, offset) );
#endif