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

Fix GPU profiling regions (#1032)

Browse Source 
Fix GPU profiling regions at `par_gauss_elim.c`
This commit is contained in:
Victor A. P. Magri 2023-12-12 17:58:08 -05:00 committed by GitHub
parent bda610a45a
commit 0dfe0fee56
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 42 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/parcsr_ls/par_gauss_elim.c
View File

@ -272,6 +272,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
{ {
/* Skip setup if this rank has no rows. */ /* Skip setup if this rank has no rows. */
hypre_ParAMGDataGSSetup(amg_data) = 1; hypre_ParAMGDataGSSetup(amg_data) = 1;
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
} }
@ -320,6 +325,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
/* Skip setup if this rank has no rows. */ /* Skip setup if this rank has no rows. */
hypre_ParAMGDataGSSetup(amg_data) = 1; hypre_ParAMGDataGSSetup(amg_data) = 1;
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -331,6 +341,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
if (!num_rows) if (!num_rows)
{ {
hypre_ParAMGDataGSSetup(amg_data) = 1; hypre_ParAMGDataGSSetup(amg_data) = 1;
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -357,6 +372,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
if (ierr != 0) if (ierr != 0)
{ {
hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetrf!"); hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetrf!");
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -373,6 +393,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
if (ierr != 0) if (ierr != 0)
{ {
hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetri (query)!"); hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetri (query)!");
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -386,6 +411,11 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
if (ierr != 0) if (ierr != 0)
{ {
hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetri!"); hypre_error_w_msg(HYPRE_ERROR_GENERIC, "Problem with dgetri!");
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
hypre_TFree(work, HYPRE_MEMORY_HOST); hypre_TFree(work, HYPRE_MEMORY_HOST);
@ -399,11 +429,12 @@ hypre_GaussElimSetup(hypre_ParAMGData *amg_data,
hypre_ParAMGDataGSSetup(amg_data) = 1; hypre_ParAMGDataGSSetup(amg_data) = 1;
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
#ifdef HYPRE_PROFILE #ifdef HYPRE_PROFILE
hypre_profile_times[HYPRE_TIMER_ID_GS_ELIM_SETUP] += hypre_MPI_Wtime(); hypre_profile_times[HYPRE_TIMER_ID_GS_ELIM_SETUP] += hypre_MPI_Wtime();
#endif #endif
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -512,6 +543,10 @@ hypre_GaussElimSolve(hypre_ParAMGData *amg_data,
hypre_TFree(b_data_h, HYPRE_MEMORY_HOST); hypre_TFree(b_data_h, HYPRE_MEMORY_HOST);
} }
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -572,6 +607,10 @@ hypre_GaussElimSolve(hypre_ParAMGData *amg_data,
hypre_TFree(b_data_h, HYPRE_MEMORY_HOST); hypre_TFree(b_data_h, HYPRE_MEMORY_HOST);
} }
/* Finalize profiling */
hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END;
return hypre_error_flag; return hypre_error_flag;
} }
@ -641,6 +680,7 @@ hypre_GaussElimSolve(hypre_ParAMGData *amg_data,
hypre_TFree(b_data_h, HYPRE_MEMORY_HOST); hypre_TFree(b_data_h, HYPRE_MEMORY_HOST);
} }
/* Finalize profiling */
hypre_GpuProfilingPopRange(); hypre_GpuProfilingPopRange();
HYPRE_ANNOTATE_FUNC_END; HYPRE_ANNOTATE_FUNC_END;
#ifdef HYPRE_PROFILE #ifdef HYPRE_PROFILE