未验证 提交 fd56f08e 编写于 作者: K kangguangli 提交者: GitHub

[Bug Fix] fix compile error in gcc540 (#45702)

* fix compile error in gcc540
上级 4b41edd6
...@@ -48,14 +48,30 @@ inline OneDNNMemoryFormat ToOneDNNFormat(const DataLayout& layout) { ...@@ -48,14 +48,30 @@ inline OneDNNMemoryFormat ToOneDNNFormat(const DataLayout& layout) {
} }
} }
// Caution: proto::VarType::Type -> phi::DataType after transfer
inline OneDNNDataType ToOneDNNDataType(DataType type) { inline OneDNNDataType ToOneDNNDataType(DataType type) {
static std::unordered_map<DataType, OneDNNDataType> dict{ #if __GNUC__ > 5
{DataType::FLOAT32, OneDNNDataType::f32}, using DataTypeMapping = std::unordered_map<DataType, OneDNNDataType>;
#else
struct DataTypeHash {
std::size_t operator()(const DataType& f) const {
return std::hash<int>{}(static_cast<int>(f));
}
};
struct DataTypeEqual {
bool operator()(const DataType& lhs, const DataType& rhs) const {
return static_cast<int>(lhs) == static_cast<int>(rhs);
}
};
using DataTypeMapping =
std::unordered_map<DataType, OneDNNDataType, DataTypeHash, DataTypeEqual>;
#endif
static DataTypeMapping dict{{DataType::FLOAT32, OneDNNDataType::f32},
{DataType::INT8, OneDNNDataType::s8}, {DataType::INT8, OneDNNDataType::s8},
{DataType::UINT8, OneDNNDataType::u8}, {DataType::UINT8, OneDNNDataType::u8},
{DataType::INT32, OneDNNDataType::s32}, {DataType::INT32, OneDNNDataType::s32},
{DataType::BFLOAT16, OneDNNDataType::bf16}}; {DataType::BFLOAT16, OneDNNDataType::bf16}};
auto iter = dict.find(type); auto iter = dict.find(type);
if (iter != dict.end()) return iter->second; if (iter != dict.end()) return iter->second;
return OneDNNDataType::undef; return OneDNNDataType::undef;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册