已合并
修改910_95数据类型 #2315
BoyangZhang创建于 4月20日
修改910_95数据类型 #2315
已合并
共 1 个文件变更+2-7
| @@ -46,15 +46,10 @@ static const std::initializer_list<DataType> ASCEND910_OPERATOR_SUPPORT_LIST = { | |||
| 46 | static const std::initializer_list<DataType> ASCEND910B_OPERATOR_SUPPORT_LIST = { | 46 | static const std::initializer_list<DataType> ASCEND910B_OPERATOR_SUPPORT_LIST = { |
| 47 | DataType::DT_FLOAT, DataType::DT_FLOAT16, DataType::DT_DOUBLE, DataType::DT_BF16}; | 47 | DataType::DT_FLOAT, DataType::DT_FLOAT16, DataType::DT_DOUBLE, DataType::DT_BF16}; |
| 48 | 48 | ||
| 49 | -static const std::initializer_list<DataType> ARCH3510_DTYPE_SUPPORT_LIST = { | ||
| 50 | - DataType::DT_FLOAT, DataType::DT_FLOAT16, DataType::DT_BF16}; | ||
| 51 | - | ||
| 52 | static inline const std::initializer_list<DataType>& GetDtypeSupportList(bool isInput) { | 49 | static inline const std::initializer_list<DataType>& GetDtypeSupportList(bool isInput) { |
| 53 | - if (GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910B || | 50 | + if (GetCurrentPlatformInfo().GetSocVersion() >= SocVersion::ASCEND910B && |
| 54 | - GetCurrentPlatformInfo().GetSocVersion() == SocVersion::ASCEND910_93) { | 51 | + GetCurrentPlatformInfo().GetSocVersion() <= SocVersion::ASCEND910E) { |
| 55 | return isInput ? ASCEND910B_INPUT_DTYPE_SUPPORT_LIST : ASCEND910B_OPERATOR_SUPPORT_LIST; | 52 | return isInput ? ASCEND910B_INPUT_DTYPE_SUPPORT_LIST : ASCEND910B_OPERATOR_SUPPORT_LIST; |
| 56 | - } else if (IsRegBase()) { | ||
| 57 | - return ARCH3510_DTYPE_SUPPORT_LIST; | ||
| 58 | } else { | 53 | } else { |
| 59 | return isInput ? ASCEND910_INPUT_DTYPE_SUPPORT_LIST : ASCEND910_OPERATOR_SUPPORT_LIST; | 54 | return isInput ? ASCEND910_INPUT_DTYPE_SUPPORT_LIST : ASCEND910_OPERATOR_SUPPORT_LIST; |
| 60 | } | 55 | } |