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

Merge pull request #450 from hypre-space/RL_FIX

Browse Source 
bigint fix
This commit is contained in:
Ruipeng Li 2021-08-10 15:26:47 -07:00 committed by GitHub
commit 3f86950f0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/parcsr_ls/par_interp_device.c
View File

@ -1246,7 +1246,7 @@ hypre_BoomerAMGBuildInterpOnePntDevice( hypre_ParCSRMatrix *A,
/* get a mapping from P offd indices to A offd indices */
/* offd_map_P_to_A[ P offd idx ] = A offd idx */
HYPRE_Int *offd_map_P_to_A = hypre_CTAlloc(HYPRE_BigInt, num_cols_P_offd, HYPRE_MEMORY_DEVICE);
HYPRE_Int *offd_map_P_to_A = hypre_CTAlloc(HYPRE_Int, num_cols_P_offd, HYPRE_MEMORY_DEVICE);
HYPRE_THRUST_CALL( copy_if,
thrust::make_counting_iterator(0),
thrust::make_counting_iterator(num_cols_A_offd),
@ -1285,7 +1285,7 @@ hypre_BoomerAMGBuildInterpOnePntDevice( hypre_ParCSRMatrix *A,
offd_map_P_to_A + num_cols_P_offd,
fine_to_coarse_offd,
col_map_offd_P_device);
hypre_TMemcpy(col_map_offd_P, col_map_offd_P_device, HYPRE_Int, num_cols_P_offd, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE);
hypre_TMemcpy(col_map_offd_P, col_map_offd_P_device, HYPRE_BigInt, num_cols_P_offd, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE);
hypre_TFree(offd_map_P_to_A, HYPRE_MEMORY_DEVICE);
hypre_TFree(col_map_offd_P_device, HYPRE_MEMORY_DEVICE);

4
src/parcsr_ls/par_lr_restr_device.c
View File

@ -265,8 +265,8 @@ hypre_BoomerAMGBuildRestrNeumannAIRDevice( hypre_ParCSRMatrix *A,
n_cpts, Fmap, Cmap, Z_diag_i, Z_diag_j, Z_diag_a, R_diag_i, R_diag_j, R_diag_a);
num_cols_offd_R = num_cols_offd_Z;
col_map_offd_R = hypre_TAlloc(HYPRE_Int, num_cols_offd_Z, HYPRE_MEMORY_HOST);
hypre_TMemcpy(col_map_offd_R, Fmap_offd_global, HYPRE_Int, num_cols_offd_Z, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE);
col_map_offd_R = hypre_TAlloc(HYPRE_BigInt, num_cols_offd_Z, HYPRE_MEMORY_HOST);
hypre_TMemcpy(col_map_offd_R, Fmap_offd_global, HYPRE_BigInt, num_cols_offd_Z, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE);
/* Now, we should have everything of Parcsr matrix R */
R = hypre_ParCSRMatrixCreate(comm,