已合并
adapt runtime struct parameter rename: localMemorySize to dynUBufSize #920
ChenZhoujie创建于 3月16日
adapt runtime struct parameter rename: localMemorySize to dynUBufSize #920
已合并
共 1 个文件变更+2-2
| @@ -243,9 +243,9 @@ int32_t AscendLaunchKernelWithHostArgs(void* funcHandle, | |||
| 243 | if (currentSoc == SocVersion::ASCEND950 && ubufDynamicSize !=0) { | 243 | if (currentSoc == SocVersion::ASCEND950 && ubufDynamicSize !=0) { |
| 244 | constexpr uint32_t attrLen = 1; | 244 | constexpr uint32_t attrLen = 1; |
| 245 | aclrtLaunchKernelAttrValue attrValue{}; | 245 | aclrtLaunchKernelAttrValue attrValue{}; |
| 246 | - attrValue.localMemorySize = ubufDynamicSize; | 246 | + attrValue.dynUBufSize = ubufDynamicSize; |
| 247 | aclrtLaunchKernelAttr attrList[attrLen] = { | 247 | aclrtLaunchKernelAttr attrList[attrLen] = { |
| 248 | - {static_cast<aclrtLaunchKernelAttrId>(2)/* ACL_RT_LAUNCH_KERNEL_ATTR_LOCAL_MEMORY_SIZE */, attrValue}, | 248 | + {static_cast<aclrtLaunchKernelAttrId>(2)/* ACL_RT_LAUNCH_KERNEL_ATTR_DYN_UBUF_SIZE */, attrValue}, |
| 249 | }; | 249 | }; |
| 250 | aclrtLaunchKernelCfg cfg = {attrList, attrLen}; | 250 | aclrtLaunchKernelCfg cfg = {attrList, attrLen}; |
| 251 | return aclrtLaunchKernelWithHostArgs(funcHandle, numBlocks, stream, &cfg, hostArgs, argsSize, nullptr, 0); | 251 | return aclrtLaunchKernelWithHostArgs(funcHandle, numBlocks, stream, &cfg, hostArgs, argsSize, nullptr, 0); |