已合并
aclnnFloorDiv A3上int64数据类型支持走AICORE分支 #1608
sunchun创建于 3月12日
aclnnFloorDiv A3上int64数据类型支持走AICORE分支 #1608
已合并
从已删除 :floordiv1合入到cann/ops-mathmaster
共 1 个文件变更+2-1
| @@ -35,7 +35,8 @@ static inline const std::initializer_list<op::DataType>& GetAiCoreDtypeSupportLi | |||
| 35 | { | 35 | { |
| 36 | auto npuArch = op::GetCurrentPlatformInfo().GetCurNpuArch(); | 36 | auto npuArch = op::GetCurrentPlatformInfo().GetCurNpuArch(); |
| 37 | switch (npuArch) { | 37 | switch (npuArch) { |
| 38 | - case NpuArch::DAV_3510: { | 38 | + case NpuArch::DAV_3510: |
| 39 | + case NpuArch::DAV_2201: { | ||
| 39 | return REGBASE_DTYPE_SUPPORT_LIST; | 40 | return REGBASE_DTYPE_SUPPORT_LIST; |
| 40 | } | 41 | } |
| 41 | default: { | 42 | default: { |