def test_trace_no_packet_context(self):
self._write_trace()
traces = btr.TraceCollection()
trace_handle = traces.add_trace(self._trace_path, 'ctf')
self.assertIsNotNone(trace_handle)
event_count = sum(1 for event in traces.events)
self.assertEqual(self._expected_event_count, event_count)
def _check_trace_expected_timestamps(trace_paths, expected_timestamps):
traces = btr.TraceCollection(intersect_mode=True)
for trace_path in trace_paths:
trace_handle = traces.add_trace(trace_path, 'ctf')
if trace_handle is None:
print('# Failed to open trace at {}'.format(trace_path))
return False
for event in traces.events:
expected_timestamp = expected_timestamps.pop(0)
if event.timestamp != expected_timestamp:
print('# Unexpected timestamp ({}), expected {}'.format(
event.timestamp, expected_timestamp))
return False
return True
from utils import debugPrint
import math
# parse argument to get the program name and path
parser = argparse.ArgumentParser(description="Sort events")
parser.add_argument("--input_trace", help="set the input trace")
parser.add_argument("--output_trace", help="set the output trace destination")
parser.add_argument(
"--gpu_log",
help=
"log file created by HC with GPU information (kernels, barriers, memcpy)")
args = parser.parse_args()
debugPrint("SORT EVENTS")
# Add the input trace to the collection
collection = btr.TraceCollection()
# Set the input traces
if args.input_trace == None:
directory = os.getcwd() + "/../lttng-traces/"
path = max([os.path.join(directory, d) for d in os.listdir(directory)],
key=os.path.getmtime)
else:
path = args.input_trace
if path[-1] == "/":
path = path[:-1]
collection.add_trace(path + "/ust/uid/1000/64-bit", 'ctf')
# Set the output trace
if args.output_trace == None:
out_path = "/tmp/tensorflow-profiler"
