已合并
foreach_add_listV2算子tiling bug修复 #7398
zhaozhongyao创建于 7月10日
foreach_add_listV2算子tiling bug修复 #7398
已合并
共 1 个文件变更+1-0
| @@ -135,6 +135,7 @@ static ge::graphStatus ForeachAddListTilingFunc(gert::TilingContext* context) | |||
| 135 | 135 | ||
| 136 | int64_t perCoreElements = Ops::Base::CeilDiv(totalElements, needCoreNum); | 136 | int64_t perCoreElements = Ops::Base::CeilDiv(totalElements, needCoreNum); |
| 137 | perCoreElements = Ops::Base::CeilDiv(perCoreElements, ALIGN_SIZE) * ALIGN_SIZE; | 137 | perCoreElements = Ops::Base::CeilDiv(perCoreElements, ALIGN_SIZE) * ALIGN_SIZE; |
| 138 | + needCoreNum = Ops::Base::CeilDiv(totalElements, perCoreElements); | ||
| 138 | 139 | ||
| 139 | int64_t globalOffset = 0; | 140 | int64_t globalOffset = 0; |
| 140 | for (int64_t core = 0; core < needCoreNum; core++) { | 141 | for (int64_t core = 0; core < needCoreNum; core++) { |