已合并
release arch restriction for 310p aclnnMatmulWeightNz #5105
llqx-1创建于 5月21日
release arch restriction for 310p aclnnMatmulWeightNz #5105
已合并
共 1 个文件变更+2-1
| @@ -281,7 +281,8 @@ bool CheckWeightNzShapeValid(const aclTensor* self, const aclTensor* mat2) | |||
| 281 | { | 281 | { |
| 282 | auto socVersion = GetCurrentPlatformInfo().GetSocVersion(); | 282 | auto socVersion = GetCurrentPlatformInfo().GetSocVersion(); |
| 283 | auto npuArch = op::GetCurrentPlatformInfo().GetCurNpuArch(); | 283 | auto npuArch = op::GetCurrentPlatformInfo().GetCurNpuArch(); |
| 284 | - bool isSupportNpuArch = ((npuArch == NpuArch::DAV_2201) || (npuArch == NpuArch::DAV_3510)); | 284 | + bool isSupportNpuArch = ((npuArch == NpuArch::DAV_2201) || (npuArch == NpuArch::DAV_3510) || |
| 285 | + (npuArch == NpuArch::DAV_2002)); | ||
| 285 | if (!isSupportNpuArch) { | 286 | if (!isSupportNpuArch) { |
| 286 | OP_LOGE( | 287 | OP_LOGE( |
| 287 | ACLNN_ERR_PARAM_INVALID, "Weight NZ is unsupported by the current SOC version [%s].", | 288 | ACLNN_ERR_PARAM_INVALID, "Weight NZ is unsupported by the current SOC version [%s].", |