mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!39673 clean code 

Merge pull request !39673 from zhupuxu/clean_master
This commit is contained in:
i-robot 2022-08-08 08:51:16 +00:00 committed by Gitee
parent a028a42c88 ef6f9a19f3
commit 2897d0dcfd
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/common/session/session_basic.cc
View File

@ -3070,7 +3070,7 @@ void SessionBasic::DumpGraphs(const std::vector<KernelGraphPtr> &graphs) const {
void SessionBasic::UnifyMindIR(const KernelGraphPtr &graph) { opt::CommonUnifyMindIR(graph); }
#ifdef WITH_BACKEND
void SessionBasic::InitPsWorker(const KernelGraphPtr &kernel_graph) {
void SessionBasic::InitPsWorker(const KernelGraphPtr &kernel_graph) const {
if (!ps::PSContext::instance()->is_worker()) {
return;
}

2
mindspore/ccsrc/backend/common/session/session_basic.h
View File

@ -351,7 +351,7 @@ class BACKEND_EXPORT SessionBasic : public std::enable_shared_from_this<SessionB
#ifdef WITH_BACKEND
void CheckPSModeConsistence(const KernelGraphPtr &kernel_graph) const;
void GetBatchElements(const AnfNodePtr &kernel_node) const;
void InitPsWorker(const KernelGraphPtr &kernel_graph);
void InitPsWorker(const KernelGraphPtr &kernel_graph) const;
#endif
// TODO(caifubi): refactor and remove bucket.
std::map<uint32_t, std::vector<std::shared_ptr<device::Bucket>>> bucket_map_;