未验证 提交 84c00546 编写于 作者: C chengduo 提交者: GitHub

Make timeline_py compatible (#16185)

* make timeline.py compatible
test=develop

* Follow comment
test=develop
上级 4ef6f738
...@@ -160,6 +160,8 @@ class Timeline(object): ...@@ -160,6 +160,8 @@ class Timeline(object):
self._devices[(k, event.device_id, "GPUKernel")] = pid self._devices[(k, event.device_id, "GPUKernel")] = pid
self._chrome_trace.emit_pid("%s:gpu:%d" % self._chrome_trace.emit_pid("%s:gpu:%d" %
(k, event.device_id), pid) (k, event.device_id), pid)
if not hasattr(profile_pb, "mem_events"):
continue
for mevent in profile_pb.mem_events: for mevent in profile_pb.mem_events:
if mevent.place == profiler_pb2.MemEvent.CUDAPlace: if mevent.place == profiler_pb2.MemEvent.CUDAPlace:
if (k, mevent.device_id, "GPU") not in self._mem_devices: if (k, mevent.device_id, "GPU") not in self._mem_devices:
...@@ -211,7 +213,7 @@ class Timeline(object): ...@@ -211,7 +213,7 @@ class Timeline(object):
args = {'name': event.name} args = {'name': event.name}
if event.memcopy.bytes > 0: if event.memcopy.bytes > 0:
args['mem_bytes'] = event.memcopy.bytes args['mem_bytes'] = event.memcopy.bytes
if event.detail_info: if hasattr(event, "detail_info") and event.detail_info:
args['detail_info'] = event.detail_info args['detail_info'] = event.detail_info
# TODO(panyx0718): Chrome tracing only handles ms. However, some # TODO(panyx0718): Chrome tracing only handles ms. However, some
# ops takes micro-seconds. Hence, we keep the ns here. # ops takes micro-seconds. Hence, we keep the ns here.
...@@ -220,6 +222,8 @@ class Timeline(object): ...@@ -220,6 +222,8 @@ class Timeline(object):
event.sub_device_id, 'Op', event.name, args) event.sub_device_id, 'Op', event.name, args)
def _allocate_memory_event(self): def _allocate_memory_event(self):
if not hasattr(profiler_pb2, "MemEvent"):
return
place_to_str = { place_to_str = {
profiler_pb2.MemEvent.CPUPlace: "CPU", profiler_pb2.MemEvent.CPUPlace: "CPU",
profiler_pb2.MemEvent.CUDAPlace: "GPU", profiler_pb2.MemEvent.CUDAPlace: "GPU",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册