提交 91d24c5f 编写于 作者: H hedaoyuan

Bug fix.

上级 3415e264
...@@ -2065,8 +2065,8 @@ public: ...@@ -2065,8 +2065,8 @@ public:
}; };
class SharedCpuMatrix : public CpuMatrix { class SharedCpuMatrix : public CpuMatrix {
#ifndef PADDLE_MOBILE_INFERENCE
public: public:
#ifndef PADDLE_MOBILE_INFERENCE
/* blockNum is number of partitions of the matrix */ /* blockNum is number of partitions of the matrix */
SharedCpuMatrix(int blockNum, size_t height, size_t width, bool trans = false) SharedCpuMatrix(int blockNum, size_t height, size_t width, bool trans = false)
: CpuMatrix(height, width, trans) { : CpuMatrix(height, width, trans) {
......
...@@ -202,6 +202,7 @@ void Parameter::setMat(ParameterType pType, int matType) { ...@@ -202,6 +202,7 @@ void Parameter::setMat(ParameterType pType, int matType) {
} }
} }
#ifndef PADDLE_MOBILE_INFERENCE #ifndef PADDLE_MOBILE_INFERENCE
// NOLINTNEXTLINE
else if (matType == MAT_NORMAL_SHARED) { else if (matType == MAT_NORMAL_SHARED) {
CHECK_EQ(height * width, bufs_[pType]->getSize()); CHECK_EQ(height * width, bufs_[pType]->getSize());
size_t blockNum = 0; size_t blockNum = 0;
...@@ -263,6 +264,7 @@ void Parameter::setMat(ParameterType pType, int matType) { ...@@ -263,6 +264,7 @@ void Parameter::setMat(ParameterType pType, int matType) {
mats_[pType] = std::make_shared<SparseAutoGrowRowCpuMatrix>(height, width); mats_[pType] = std::make_shared<SparseAutoGrowRowCpuMatrix>(height, width);
} }
#endif #endif
// NOLINTNEXTLINE
else { else {
LOG(FATAL) << "Unsupported mat type" << matType; LOG(FATAL) << "Unsupported mat type" << matType;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册