Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!8975 Modify the permisson of gpu profiler file and directory 

From: @gzhcv
Reviewed-by: @wangyue01,@yuximiao
Signed-off-by: @yuximiao
This commit is contained in:
mindspore-ci-bot 2020-11-25 14:35:39 +08:00 committed by Gitee
parent 2bdc6198aa 35996eafcf
commit c84253d9cd
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/profiler/device/gpu/data_saver.cc
View File

@ -310,7 +310,7 @@ void DataSaver::WriteStepTrace(const std::string &saver_base_dir) {
void DataSaver::SetStepTraceOpName(ProfilingTraceInfo trace_op_name) { step_trace_op_name = trace_op_name; } void DataSaver::SetStepTraceOpName(ProfilingTraceInfo trace_op_name) { step_trace_op_name = trace_op_name; }
void DataSaver::ChangeFileMode(const std::string &file_path) { void DataSaver::ChangeFileMode(const std::string &file_path) {
if (chmod(common::SafeCStr(file_path), S_IRUSR | S_IWUSR) == -1) { if (chmod(common::SafeCStr(file_path), S_IRUSR) == -1) {
MS_LOG(WARNING) << "Modify file:" << file_path << " to rw fail."; MS_LOG(WARNING) << "Modify file:" << file_path << " to rw fail.";
return; return;
} }

6
mindspore/profiler/parser/step_trace_parser.py
View File

@ -329,7 +329,7 @@ class BaseStepTraceParser:
csv_writer.writerow(self._header) csv_writer.writerow(self._header)
for row_data in self._result: for row_data in self._result:
csv_writer.writerow(row_data) csv_writer.writerow(row_data)
os.chmod(self._output_path, stat.S_IREAD) os.chmod(self._output_path, stat.S_IRUSR)
class GpuStepTraceParser(BaseStepTraceParser): class GpuStepTraceParser(BaseStepTraceParser):
@ -363,7 +363,7 @@ class GpuStepTraceParser(BaseStepTraceParser):
try: try:
with open(output_path, 'w') as json_file: with open(output_path, 'w') as json_file:
json.dump(points, json_file) json.dump(points, json_file)
os.chmod(output_path, stat.S_IREAD) os.chmod(output_path, stat.S_IRUSR)
except (IOError, OSError) as err: except (IOError, OSError) as err:
log.warning('Failed to save point info. %s', err) log.warning('Failed to save point info. %s', err)
raise ProfilerIOException raise ProfilerIOException
@ -463,7 +463,7 @@ class AscendStepTraceParser(BaseStepTraceParser):
try: try:
with open(output_path, 'w') as json_file: with open(output_path, 'w') as json_file:
json.dump(points, json_file) json.dump(points, json_file)
os.chmod(output_path, stat.S_IREAD) os.chmod(output_path, stat.S_IRUSR)
except (IOError, OSError) as err: except (IOError, OSError) as err:
log.warning('Failed to save point info. %s', err) log.warning('Failed to save point info. %s', err)
raise ProfilerIOException raise ProfilerIOException

2
mindspore/profiler/profiling.py
View File

@ -14,6 +14,7 @@
# ============================================================================ # ============================================================================
"""Profiling api file.""" """Profiling api file."""
import os import os
import stat
import time import time
from enum import Enum from enum import Enum
@ -84,6 +85,7 @@ class Profiler:
self._output_path = os.path.join(self._output_path, "profiler") self._output_path = os.path.join(self._output_path, "profiler")
if not os.path.exists(self._output_path): if not os.path.exists(self._output_path):
os.makedirs(self._output_path, exist_ok=True) os.makedirs(self._output_path, exist_ok=True)
os.chmod(self._output_path, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
else: else:
logger.warning("The target dir already exists. " logger.warning("The target dir already exists. "
"There may be some old profiling data, and they will be rewrote in the end.") "There may be some old profiling data, and they will be rewrote in the end.")