Trace general

utils/babeltrace_thapi.in

@@ -159,6 +159,20 @@ def get_components(names)
       require 'babeltrace_ze_lib' if $options[:backends].include?('ze')
       require 'babeltrace_cuda_lib' if $options[:backends].include?('cuda')
       require 'babeltrace_hip_lib' if $options[:backends].include?('hip')
+      # I guess need to put it in `babeltrace_energy_lib` at some point?
+      $energies={}
+      $event_lambdas["lttng_ust_ze_sampling:gpu_energy"] = lambda { |defi|
+        energy = defi['energy']
+        timestamp = defi['timestamp']
+        key = [ defi['hDevice'], defi['domain'] ]
+        previous = $energies[key]
+        if previous
+          p_energy, p_timestamp = previous
+          puts "#{key[0]}:#{key[1]}: #{(energy - p_energy).to_f/(timestamp - p_timestamp)}"
+        end
+        $energies[key] = [energy, timestamp]
+      }
+
       f = lambda { |iterator, _|
         iterator.next_messages.each do |m|
           next unless m.type == :BT_MESSAGE_TYPE_EVENT