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

Bug fix.

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