Browse Source

!2957 fix stream for memory reuse

Merge pull request !2957 from gukecai/fix-stream-bug
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d2809954fd
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/device/ascend/ascend_stream_assign.cc

+ 1
- 1
mindspore/ccsrc/device/ascend/ascend_stream_assign.cc View File

@@ -1106,7 +1106,7 @@ StreamActiveKind AscendStreamAssign::GetStreamActiveKind(const NotNull<KernelGra

uint32_t pre_stream_id = UINT32_MAX;
uint32_t next_stream_id = UINT32_MAX;
int32_t start = SizeToInt(index);
int32_t start = SizeToInt(index) - 1;
for (int32_t i = start; i >= 0; i--) {
auto cnode = exe_orders[IntToSize(i)];
auto name = AnfAlgo::GetCNodeName(cnode);


Loading…
Cancel
Save