提交 ba4322f4 编写于 作者: Q Qiao Longfei 提交者: QI JUN

fix clang error (#7084)

上级 3a93fa77
...@@ -31,18 +31,16 @@ using namespace platform; ...@@ -31,18 +31,16 @@ using namespace platform;
* 1111 -> FP64, GPUPlace, kNCHW, kMKLDNN * 1111 -> FP64, GPUPlace, kNCHW, kMKLDNN
*/ */
std::array<proto::DataType, 2> kDataType = {proto::DataType::FP32, std::array<proto::DataType, 2> kDataType = {
proto::DataType::FP64}; {proto::DataType::FP32, proto::DataType::FP64}};
std::array<Place, 2> kPlace = {CPUPlace(), CUDAPlace(0)}; std::array<Place, 2> kPlace = {{CPUPlace(), CUDAPlace(0)}};
std::array<DataLayout, 2> kDataLayout = { std::array<DataLayout, 2> kDataLayout = {
DataLayout::kNHWC, DataLayout::kNCHW, {DataLayout::kNHWC, DataLayout::kNCHW}};
};
std::array<LibraryType, 2> kLibraryType = { std::array<LibraryType, 2> kLibraryType = {
LibraryType::kPlain, LibraryType::kMKLDNN, {LibraryType::kPlain, LibraryType::kMKLDNN}};
};
OpKernelType GenFromBit(const std::vector<bool> bits) { OpKernelType GenFromBit(const std::vector<bool> bits) {
return OpKernelType(kDataType[bits[0]], kPlace[bits[1]], kDataLayout[bits[2]], return OpKernelType(kDataType[bits[0]], kPlace[bits[1]], kDataLayout[bits[2]],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册