已合并
增加mselossv2损失函数核间同步 #3997
佐助大王创建于 4月20日
增加mselossv2损失函数核间同步 #3997
已合并
共 1 个文件变更+1-0
| @@ -110,6 +110,7 @@ static ge::graphStatus GetTilingAttr(gert::TilingContext* context) | |||
| 110 | context->SetScheduleMode(BATCH_MODE); | 110 | context->SetScheduleMode(BATCH_MODE); |
| 111 | } else if (optiling::reduction == optiling::REDUCTION_MEAN) { | 111 | } else if (optiling::reduction == optiling::REDUCTION_MEAN) { |
| 112 | optiling::tilingKey += TILING_KEY_ATTR_MEAN; | 112 | optiling::tilingKey += TILING_KEY_ATTR_MEAN; |
| 113 | + context->SetScheduleMode(BATCH_MODE); | ||
| 113 | } else { | 114 | } else { |
| 114 | OP_LOGD(context, "The reduction attribute must be 'none', 'mean', or 'sum'."); | 115 | OP_LOGD(context, "The reduction attribute must be 'none', 'mean', or 'sum'."); |
| 115 | return ge::GRAPH_FAILED; | 116 | return ge::GRAPH_FAILED; |