This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
bug fix for scalar input HCCL ops
tags/v1.2.0-rc1
alouhahaha
4 years ago
parent
fbcfbed66d
commit
580138f1fe
1 changed files
with
3 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+3
-1
mindspore/ccsrc/backend/optimizer/pass/communication_op_fusion.cc
+ 3
- 1
mindspore/ccsrc/backend/optimizer/pass/communication_op_fusion.cc
View File
@@ -204,7 +204,9 @@ AnfNodePtr CommunicationOpFusion::CreateFusedCommunicationOp(const FuncGraphPtr
auto cnode = communication_op_info.communication_op_nodes[idx];
MS_EXCEPTION_IF_NULL(cnode);
std::vector<size_t> shape = AnfAlgo::GetOutputInferShape(cnode, 0);
shape[0] /= rank_size;
if (!shape.empty()) {
shape[0] /= rank_size;
}
shapes.push_back(shape);
}
}
Write
Preview
Loading…
Cancel
Save