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

bug fix

Browse Source
This commit is contained in:
Rui-peng Li 2022-03-10 22:19:22 -06:00
parent 9498625db4
commit 700b0328bf
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