From 09b35e11432d357b4d0879fdd80b3f9824aa3390 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Tue, 26 Jan 2021 22:59:44 +0100
Subject: [PATCH] Fix enable/disable traces

---
 src/testCHAMELEON.c  | 12 ++++++++++++
 src/testHCHAMELEON.c | 10 ++++++++--
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/testCHAMELEON.c b/src/testCHAMELEON.c
index 82b2971..6314823 100644
--- a/src/testCHAMELEON.c
+++ b/src/testCHAMELEON.c
@@ -79,7 +79,13 @@ int testCHAMELEON(void) {
                                     nbPts, nbRHS, 1, np ); CHKERRQ(ierr);
 
       /* Compute the mat.vec product A.rhs -> solCHAM */
+      if ( generate_traces ) {
+        CHAMELEON_Enable( CHAMELEON_PROFILING_MODE );
+      }
       CHAMELEON_gemm_Tile( descA, descX, descY );
+      if ( generate_traces ) {
+        CHAMELEON_Disable( CHAMELEON_PROFILING_MODE );
+      }
 
       temps_final = getTime ();
       temps_cpu = temps_final - temps_initial ;
@@ -117,6 +123,9 @@ int testCHAMELEON(void) {
       /* Factorize the CHAMELEON Matrix */
       /* ===================================================================================== */
       Mpf_printf(MPI_COMM_WORLD, "\n\n**** Factorizing CHAMELEON Mat...\n") ;
+      if ( generate_traces ) {
+        CHAMELEON_Enable( CHAMELEON_PROFILING_MODE );
+      }
       temps_initial = getTime ();
       if (symMatSolver) {
         ierr = CHAMELEON_sytrf_Tile (ChamLower, descA); CHKERRQ(ierr);
@@ -125,6 +134,9 @@ int testCHAMELEON(void) {
       }
       temps_final = getTime ();
       temps_cpu = temps_final - temps_initial ;
+      if ( generate_traces ) {
+        CHAMELEON_Disable( CHAMELEON_PROFILING_MODE );
+      }
       if ( verboseTest ) {
         void *lapackA = (void*)MpfCalloc( nbPts * nbPts, sizeof(D_type) ) ; CHKPTRQ(lapackA);
 
diff --git a/src/testHCHAMELEON.c b/src/testHCHAMELEON.c
index 86b925e..89c85ec 100644
--- a/src/testHCHAMELEON.c
+++ b/src/testHCHAMELEON.c
@@ -150,7 +150,13 @@ int testHCHAMELEON(void) {
                                     nbPts, nbRHS, dims[0], dims[1] ); CHKERRQ(ierr);
 
       /* Compute the mat.vec product A.rhs -> solCHAM */
+      if ( generate_traces ) {
+        CHAMELEON_Enable( CHAMELEON_PROFILING_MODE );
+      }
       CHAMELEON_gemm_Tile( hdescA.super, descX, descY );
+      if ( generate_traces ) {
+        CHAMELEON_Disable( CHAMELEON_PROFILING_MODE );
+      }
 
       /* convert back to lapack format */
       ierr = CHAMELEON_Desc2Lap( ChamUpperLower, descY, solCHAM, nbPts ); CHKERRQ(ierr);
@@ -197,7 +203,7 @@ int testHCHAMELEON(void) {
       /* ===================================================================================== */
       Mpf_printf(MPI_COMM_WORLD, "\n\n**** Factorizing H-CHAMELEON...\n") ;
       if ( generate_traces ) {
-        CHAMELEON_Enable(CHAMELEON_PROFILING_MODE);
+        CHAMELEON_Enable( CHAMELEON_PROFILING_MODE );
       }
       temps_initial = getTime ();
       if ( symMatSolver ) {
@@ -208,7 +214,7 @@ int testHCHAMELEON(void) {
       temps_final = getTime ();
       temps_cpu = temps_final - temps_initial ;
       if ( generate_traces ) {
-        CHAMELEON_Disable(CHAMELEON_PROFILING_MODE);
+        CHAMELEON_Disable( CHAMELEON_PROFILING_MODE );
       }
       Mpf_printf(MPI_COMM_WORLD,"<PERFTESTS> TpsCpuFacto = %f \n", temps_cpu) ;
 
-- 
GitLab