From bc7c2281b1487d8c374f0c4ba9a9bc45862e3aaf Mon Sep 17 00:00:00 2001 From: gzhcv Date: Wed, 9 Sep 2020 21:56:28 +0800 Subject: [PATCH] gpu_timeline(python code) update mindspore/profiler/parser/integrator.py. update mindspore/profiler/profiling.py. --- mindspore/profiler/profiling.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/profiler/profiling.py b/mindspore/profiler/profiling.py index 5db1fa32f6..f89c177411 100644 --- a/mindspore/profiler/profiling.py +++ b/mindspore/profiler/profiling.py @@ -18,7 +18,7 @@ import time from enum import Enum from mindspore import log as logger, context -from mindspore.communication.management import release, init, get_rank +from mindspore.communication.management import release, get_rank from mindspore.profiler.common.exceptions.exceptions import ProfilerFileNotFoundException, \ ProfilerIOException, ProfilerException from mindspore.profiler.common.util import get_file_names, fwrite_format @@ -96,8 +96,8 @@ class Profiler: self._gpu_profiler = GPUProfiler.get_instance() self._gpu_profiler.init(self._output_path) self._gpu_profiler.step_profiling_enable(True) - init() - self._dev_id = get_rank() + if context.get_auto_parallel_context('device_num') > 1: + self._dev_id = get_rank() os.environ['DEVICE_ID'] = str(self._dev_id) if kwargs: