Browse Source

fix all reduce and loss overlap

pull/315/head
baker 4 years ago
parent
commit
2fe1128c07
1 changed files with 0 additions and 12 deletions
  1. +0
    -12
      ge/graph/build/logical_stream_allocator.h

+ 0
- 12
ge/graph/build/logical_stream_allocator.h View File

@@ -147,18 +147,6 @@ class NodeStreamUpdatePass : public LogicalStreamPass {
public:
STREAM_PASS_DEFAULT_FUNC(NodeStreamUpdatePass);
Status Run(ComputeGraphPtr graph, const std::vector<SubgraphPtr> &subgraphs, Context &context) override;

private:
/// Optimize for case like:
/// NodeA(stream1) -> Const(stream2) -> NodeB(stream1)
/// To case:
/// NodeA(stream1) -> Const(stream1) -> NodeB(stream1)
/// Which could reduce event number (Const could be other type which belong to skipped engine subgraph)
Status UpdateForSkippedEngine(const ComputeGraphPtr &graph, const std::vector<SubgraphPtr> &subgraphs);

int64_t GetSingleInoutStream(const NodePtr &node) const;
// Judge if all predecessors' streams of node are kInvalidStream
bool AreAllPredStreamsInvalid(const NodePtr &node) const;
};

// Update the stream of subgraphs to nodes.


Loading…
Cancel
Save