bigint fix

This commit is contained in:
Ruipeng Li 2021-08-10 13:12:40 -07:00
parent efec0bec11
commit e262a9dde1
2 changed files with 24 additions and 24 deletions

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),

View File

@ -265,7 +265,7 @@ 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);
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_Int, num_cols_offd_Z, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE);
/* Now, we should have everything of Parcsr matrix R */