Merged
fix: reduce 普通场景分配最小 workspace，避免 kernel 入口提前返回 #1142
sunzhongwen1Add on 4 days ago
fix: reduce 普通场景分配最小 workspace，避免 kernel 入口提前返回 #1142
Merged
Merged fromsunzhongwen1/opbasebugfix/workspaceintocann/opbasemaster
Total: 1 files changed+2-0
| @@ -1261,6 +1261,8 @@ void ReduceOpTiling::CalcUserWorkSpace() | |||
| 1261 | auto platformInfo = context_->GetPlatformInfo(); | 1261 | auto platformInfo = context_->GetPlatformInfo(); |
| 1262 | auto ascendcPlatform = platform_ascendc::PlatformAscendC(platformInfo); | 1262 | auto ascendcPlatform = platform_ascendc::PlatformAscendC(platformInfo); |
| 1263 | sysWorkSpaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); | 1263 | sysWorkSpaceSize = ascendcPlatform.GetLibApiWorkSpaceSize(); |
| 1264 | + } else { | ||
| 1265 | + sysWorkSpaceSize = 32; // normal场景分配32Bworkspace，防止算子kernel入口返回空指针 | ||
| 1264 | } | 1266 | } |
| 1265 | workspaces[0] = sysWorkSpaceSize + workSpaceSize_; | 1267 | workspaces[0] = sysWorkSpaceSize + workSpaceSize_; |
| 1266 | } | 1268 | } |