From 8435cfd741f8afcadae172cdc99b21f1dcf4db51 Mon Sep 17 00:00:00 2001 From: Thomas Applencourt Date: Mon, 11 Sep 2023 14:58:07 +0000 Subject: [PATCH] Trace general --- utils/babeltrace_thapi.in | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/utils/babeltrace_thapi.in b/utils/babeltrace_thapi.in index 794978e4..bae384b8 100755 --- a/utils/babeltrace_thapi.in +++ b/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