|
|
|
@@ -174,8 +174,8 @@ class FrameworkParser: |
|
|
|
output_path (str): The directory of the parsed file. Default: `./`. |
|
|
|
""" |
|
|
|
_raw_data_dir = '/var/log/npu/profiling' |
|
|
|
_regex_framework = r'Framework\.host\.(?P<data_type>.+)\.(?P<device_id>\d).+' |
|
|
|
_regex_framework_in_data = r'Framework\.host\.(?P<data_type>.+)\.' \ |
|
|
|
_regex_framework = r'Framework\.(?P<data_type>.+)\.(?P<device_id>\d).+' |
|
|
|
_regex_framework_in_data = r'Framework\.(?P<data_type>.+)\.' \ |
|
|
|
r'(?P<device_id>\d)\.(?P<profiling_id>[a-zA-Z0-9]+).+' |
|
|
|
_col_names = [ |
|
|
|
'task_id', 'stream_id', 'block_dim', 'full_op_name', 'op_name', |
|
|
|
@@ -338,6 +338,7 @@ class FrameworkParser: |
|
|
|
raise ProfilerDeviceIdMismatchException() |
|
|
|
|
|
|
|
data_type = attrs.get('data_type') |
|
|
|
data_type = data_type.replace("host.", "") |
|
|
|
if data_type.startswith('vm.'): |
|
|
|
if self._backend_type and self._backend_type != 'vm': |
|
|
|
raise ProfilerRawFileException('Backend type is inconsistent.') |
|
|
|
@@ -395,6 +396,7 @@ class FrameworkParser: |
|
|
|
raise ProfilerDeviceIdMismatchException() |
|
|
|
|
|
|
|
data_type = attrs.get('data_type') |
|
|
|
data_type = data_type.replace("host.", "") |
|
|
|
if data_type.startswith('vm.'): |
|
|
|
if self._backend_type and self._backend_type != 'vm': |
|
|
|
raise ProfilerRawFileException('Backend type is inconsistent.') |
|
|
|
|