put method csv.field_size_limit() in method __int__() of class Integrator

This commit is contained in:
z30020733 2022-12-27 10:44:24 +08:00
parent a1d5aebc72
commit dea39c2a15
1 changed files with 1 additions and 3 deletions

View File

@ -26,9 +26,6 @@ from mindspore.profiler.common.util import query_latest_trace_time_file, to_int,
from mindspore.profiler.common.validator.validate_path import validate_and_normalize_path from mindspore.profiler.common.validator.validate_path import validate_and_normalize_path
csv.field_size_limit(sys.maxsize)
class Integrator: class Integrator:
""" """
The integrator for integrating parsed profiling files. The integrator for integrating parsed profiling files.
@ -57,6 +54,7 @@ class Integrator:
_aicore_trace_data = [] _aicore_trace_data = []
def __init__(self, profiling_dir, device_id): def __init__(self, profiling_dir, device_id):
csv.field_size_limit(sys.maxsize)
self._profiling_dir = profiling_dir self._profiling_dir = profiling_dir
self._device_id = device_id self._device_id = device_id
self._op_time_cache = {} self._op_time_cache = {}