已合并
修复MoeTokenUnpermute #1081
fazhenyao123创建于 1月27日
修复MoeTokenUnpermute #1081
已合并
共 1 个文件变更+1-5
| @@ -20,10 +20,6 @@ ge::graphStatus TilingMoeTokenUnpermute(gert::TilingContext* context); | |||
| 20 | 20 | ||
| 21 | ge::graphStatus TilingMoeTokenUnpermute(gert::TilingContext* context) | 21 | ge::graphStatus TilingMoeTokenUnpermute(gert::TilingContext* context) |
| 22 | { | 22 | { |
| 23 | // 设置workspace | ||
| 24 | size_t sysWorkspaceSize = 16 * 1024 * 1024; | ||
| 25 | size_t* currentWorkspace = context->GetWorkspaceSizes(1); | ||
| 26 | currentWorkspace[0] = sysWorkspaceSize; | ||
| 27 | return TilingCompute(context, -1); | 23 | return TilingCompute(context, -1); |
| 28 | } | 24 | } |
| 29 | 25 | ||
| @@ -367,7 +363,7 @@ ge::graphStatus TilingCompute(gert::TilingContext* context, const int64_t topK) | |||
| 367 | const auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); | 363 | const auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); |
| 368 | size_t* workspaces = context->GetWorkspaceSizes(1); | 364 | size_t* workspaces = context->GetWorkspaceSizes(1); |
| 369 | workspaces[0] = ascendcPlatform.GetLibApiWorkSpaceSize(); | 365 | workspaces[0] = ascendcPlatform.GetLibApiWorkSpaceSize(); |
| 370 | 366 | OP_LOGD(context->GetNodeName(), "workspace is %zu", workspaces[0]); | |
| 371 | return context->SetTilingKey(param.core.tilingKey); | 367 | return context->SetTilingKey(param.core.tilingKey); |
| 372 | } | 368 | } |
| 373 | 369 | ||