Enable sampling energy with --sample.

argonne-lcf · Sep 8, 2023 · ea856a9 · ea856a9
1 parent 8e2b734
commit ea856a9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xprof/xprof.sh.erb.in b/xprof/xprof.sh.erb.in
@@ -336,9 +336,12 @@ setup_lttng() {
     #Activate sampling on non-blocking stream
     if [ $sample == true ]; then
         export LTTNG_UST_SAMPLING=1
+        export LTTNG_UST_SAMPLING_ENERGY=1
         lttngq enable-channel --userspace nonblocking-channel
         lttngq enable-event --channel=nonblocking-channel --userspace lttng_ust_sampling:*
+<% if languages.include?("ze") %>
         lttngq enable-event --channel=nonblocking-channel --userspace lttng_ust_ze_sampling:*
+<% end %>
         lttngq add-context --userspace --channel=nonblocking-channel -t vpid -t vtid
     fi