fix pcontrol disabling

spcl · Oct 22, 2023 · dda141f · dda141f
1 parent a90b32e
commit dda141f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/liballprof2/gencode.py b/src/liballprof2/gencode.py
@@ -300,10 +300,10 @@ def produce_pcontrol(self, mode):
         deref = ""
         if mode == 'fortran':
             deref = "*"
-        self.outfile.write(f"  if ({deref}level == 0) lap_tracing_enabled = 0;\n")
-        self.outfile.write(f"  if ({deref}level  > 0) lap_tracing_enabled = 1;\n")
-        self.outfile.write(f"  if ({deref}level  > 1) lap_elem_tracing_enabled = 1;\n")
-        self.outfile.write(f"  if ({deref}level  > 2) lap_backtrace_enabled = 1;\n")
+        self.outfile.write(f"  if ({deref}level == 0) {{ lap_tracing_enabled = 0; lap_elem_tracing_enabled = 0; lap_backtrace_enabled = 0; }}\n")
+        self.outfile.write(f"  if ({deref}level == 1) {{ lap_tracing_enabled = 1; lap_elem_tracing_enabled = 0; lap_backtrace_enabled = 0; }}\n")
+        self.outfile.write(f"  if ({deref}level == 2) {{ lap_tracing_enabled = 1; lap_elem_tracing_enabled = 1; lap_backtrace_enabled = 0; }}\n")
+        self.outfile.write(f"  if ({deref}level >= 3) {{ lap_tracing_enabled = 1; lap_elem_tracing_enabled = 1; lap_backtrace_enabled = 1; }}\n")
         if mode == 'c':
             self.outfile.write(f"  return MPI_SUCCESS;\n")