已合并
aclnnAll中950soc判断修改为arch #3002
sunzhongwen1创建于 5月28日
aclnnAll中950soc判断修改为arch #3002
已合并
共 1 个文件变更+1-1
| @@ -294,7 +294,7 @@ aclnnStatus aclnnAllGetWorkspaceSize( | |||
| 294 | 294 | ||
| 295 | auto selfCasted = selfContiguous; | 295 | auto selfCasted = selfContiguous; |
| 296 | if (!((selfContiguous->GetDataType() == op::DataType::DT_FLOAT16 || selfContiguous->GetDataType() == op::DataType::DT_BF16 || selfContiguous->GetDataType() == op::DataType::DT_FLOAT) && | 296 | if (!((selfContiguous->GetDataType() == op::DataType::DT_FLOAT16 || selfContiguous->GetDataType() == op::DataType::DT_BF16 || selfContiguous->GetDataType() == op::DataType::DT_FLOAT) && |
| 297 | - (GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910B || GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910_93 || GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND950))) { | 297 | + (GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910B || GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910_93 || GetCurrentPlatformInfo().GetCurNpuArch() == NpuArch::DAV_3510))) { |
| 298 | // 将输入self的数据类型转换成隐式数据类型，根据具体算子语义按需调用 | 298 | // 将输入self的数据类型转换成隐式数据类型，根据具体算子语义按需调用 |
| 299 | selfCasted = l0op::Cast(selfContiguous, DataType::DT_BOOL, uniqueExecutor.get()); | 299 | selfCasted = l0op::Cast(selfContiguous, DataType::DT_BOOL, uniqueExecutor.get()); |
| 300 | CHECK_RET(selfCasted != nullptr, ACLNN_ERR_INNER_NULLPTR); | 300 | CHECK_RET(selfCasted != nullptr, ACLNN_ERR_INNER_NULLPTR); |