Browse Source

fix comment

tags/v1.1.0
wangxiaotian22 3 years ago
parent
commit
c0dc27f580
2 changed files with 4 additions and 4 deletions
  1. +3
    -3
      ge/graph/passes/atomic_addr_clean_pass.cc
  2. +1
    -1
      ge/graph/passes/atomic_addr_clean_pass.h

+ 3
- 3
ge/graph/passes/atomic_addr_clean_pass.cc View File

@@ -266,7 +266,7 @@ bool AtomicAddrCleanPass::IsAtomicOp(const NodePtr &node) {
std::vector<int64_t> atomic_output_index; std::vector<int64_t> atomic_output_index;
(void) ge::AttrUtils::GetListInt(op_desc, ATOMIC_ATTR_OUTPUT_INDEX, atomic_output_index); (void) ge::AttrUtils::GetListInt(op_desc, ATOMIC_ATTR_OUTPUT_INDEX, atomic_output_index);
bool is_all_output_peer_also_atomic = true; bool is_all_output_peer_also_atomic = true;
for (auto &output_index : atomic_output_index) {
for (const auto &output_index : atomic_output_index) {
if (!IsOutputIndexPeerInputAtomic(node, output_index)) { if (!IsOutputIndexPeerInputAtomic(node, output_index)) {
is_all_output_peer_also_atomic = false; is_all_output_peer_also_atomic = false;
break; break;
@@ -318,13 +318,13 @@ bool AtomicAddrCleanPass::CheckAtomicFromOpsKernel(const NodePtr &node) {
return false; return false;
} }


bool AtomicAddrCleanPass::IsOutputIndexPeerInputAtomic(const ge::NodePtr &node, int64_t output_index) {
bool AtomicAddrCleanPass::IsOutputIndexPeerInputAtomic(const NodePtr &node, int64_t output_index) {
auto out_data_anchor = node->GetAllOutDataAnchors().at(output_index); auto out_data_anchor = node->GetAllOutDataAnchors().at(output_index);
if (out_data_anchor == nullptr) { if (out_data_anchor == nullptr) {
return false; return false;
} }


for (auto input_anchor : out_data_anchor->GetPeerInDataAnchors()) {
for (const auto input_anchor : out_data_anchor->GetPeerInDataAnchors()) {
auto output_node = input_anchor->GetOwnerNode(); auto output_node = input_anchor->GetOwnerNode();
// just hccl may mark atomic from ops kernel now, and hccl's atomic if for all input // just hccl may mark atomic from ops kernel now, and hccl's atomic if for all input
// hccl's attr ATOMIC_ATTR_INPUT_INDEX mark on CalcOpRunningParam, can't be get here // hccl's attr ATOMIC_ATTR_INPUT_INDEX mark on CalcOpRunningParam, can't be get here


+ 1
- 1
ge/graph/passes/atomic_addr_clean_pass.h View File

@@ -86,7 +86,7 @@ class AtomicAddrCleanPass : public GraphPass {


bool CheckAtomicFromOpsKernel(const NodePtr &node); bool CheckAtomicFromOpsKernel(const NodePtr &node);


bool IsOutputIndexPeerInputAtomic(const ge::NodePtr &node, int64_t output_index);
bool IsOutputIndexPeerInputAtomic(const NodePtr &node, int64_t output_index);


vector<NodePtr> hcom_node_vec_; vector<NodePtr> hcom_node_vec_;
bool is_loop_graph_ = false; bool is_loop_graph_ = false;


Loading…
Cancel
Save