fix mpi colleen

argonne-lcf · Sep 19, 2024 · 12a44d1 · 12a44d1
1 parent d3a370c
commit 12a44d1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/xprof/xprof.rb.in b/xprof/xprof.rb.in
@@ -517,7 +517,12 @@ end
 
 def enable_events_mpi(channel_name, tracing_mode: 'default', profiling: true)
   lttng_enable = "lttng enable-event --userspace --session=#{lttng_session_uuid} --channel=#{channel_name}"
-  exec("#{lttng_enable} lttng_ust_mpi:*")
+  case tracing_mode
+  when 'full'
+    exec("#{lttng_enable} lttng_ust_mpi:*")
+  when 'default'
+    exec("#{lttng_enable} lttng_ust_mpi:* -x lttng_ust_mpi:MPI_WTime*")
+  end
   exec("#{lttng_enable} lttng_ust_mpi_type:*")
 end