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

BigJ realloc (#623)

Browse Source 
This PR fixes reallocation issues with CSR BigJ arrays.
This commit is contained in:
Rui Peng Li 2023-12-01 09:46:57 -08:00 committed by GitHub
parent 76b8adc21a
commit db1364a392
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/seq_mv/csr_matrix.c
View File

@ -165,8 +165,10 @@ hypre_CSRMatrixInitialize( hypre_CSRMatrix *matrix )
*--------------------------------------------------------------------------*/
HYPRE_Int
hypre_CSRMatrixResize( hypre_CSRMatrix *matrix, HYPRE_Int new_num_rows, HYPRE_Int new_num_cols,
HYPRE_Int new_num_nonzeros )
hypre_CSRMatrixResize( hypre_CSRMatrix *matrix,
HYPRE_Int new_num_rows,
HYPRE_Int new_num_cols,
HYPRE_Int new_num_nonzeros )
{
HYPRE_MemoryLocation memory_location = hypre_CSRMatrixMemoryLocation(matrix);
HYPRE_Int old_num_nonzeros = hypre_CSRMatrixNumNonzeros(matrix);
@ -191,18 +193,26 @@ hypre_CSRMatrixResize( hypre_CSRMatrix *matrix, HYPRE_Int new_num_rows, HYPRE_In
}
else
{
hypre_CSRMatrixData(matrix) = hypre_TReAlloc_v2(hypre_CSRMatrixData(matrix), HYPRE_Complex,
old_num_nonzeros, HYPRE_Complex, new_num_nonzeros, memory_location);
hypre_CSRMatrixData(matrix) = hypre_TReAlloc_v2(hypre_CSRMatrixData(matrix), HYPRE_Complex, old_num_nonzeros,
HYPRE_Complex, new_num_nonzeros, memory_location);
}
if (!hypre_CSRMatrixJ(matrix))
if (hypre_CSRMatrixBigJ(matrix))
{
hypre_CSRMatrixJ(matrix) = hypre_CTAlloc(HYPRE_Int, new_num_nonzeros, memory_location);
hypre_CSRMatrixBigJ(matrix) = hypre_TReAlloc_v2(hypre_CSRMatrixBigJ(matrix), HYPRE_BigInt, old_num_nonzeros,
HYPRE_BigInt, new_num_nonzeros, memory_location);
}
else
{
hypre_CSRMatrixJ(matrix) = hypre_TReAlloc_v2(hypre_CSRMatrixJ(matrix), HYPRE_Int, old_num_nonzeros,
HYPRE_Int, new_num_nonzeros, memory_location);
if (!hypre_CSRMatrixJ(matrix))
{
hypre_CSRMatrixJ(matrix) = hypre_CTAlloc(HYPRE_Int, new_num_nonzeros, memory_location);
}
else
{
hypre_CSRMatrixJ(matrix) = hypre_TReAlloc_v2(hypre_CSRMatrixJ(matrix), HYPRE_Int, old_num_nonzeros,
HYPRE_Int, new_num_nonzeros, memory_location);
}
}
}