提交 0f8aad29 编写于 作者: T tensor-tang

fix compile error

上级 2e101df7
......@@ -27,7 +27,7 @@ protected:
size_t weightWidth_;
public:
MKLPackedGemm(MatrixPtr weight) {
explicit MKLPackedGemm(MatrixPtr weight) {
weightHeight_ = weight->getHeight();
weightWidth_ = weight->getWidth();
weightPacked_ =
......@@ -91,4 +91,5 @@ public:
cblas_sgemm_free(weightTPacked_);
}
};
} // namespace paddle
......@@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#pragma once
#include <gflags/gflags.h>
#include "Layer.h"
#include "MKLPackedGemm.h"
......@@ -128,4 +130,5 @@ protected:
std::unique_ptr<MKLPackedGemm> sgemm_packed_;
};
}
} // namespace paddle
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册