enable copyFrom of MKLDNNMatrix

02fdf241 · tensor-tang · ac6f0669 · 02fdf241
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

paddle/math/MKLDNNMatrix.h paddle/math/MKLDNNMatrix.h +5 -0

未找到文件。
--- a/paddle/math/MKLDNNMatrix.h
+++ b/paddle/math/MKLDNNMatrix.h
@@ -91,6 +91,11 @@ public:
      const MKLDNNMatrixPtr& dst,
      bool checkData = true);

+  void copyFrom(const Matrix& src) {
+    // TODO(TJ): reorder data if this format is not nchw or x
+    m_->copyFrom(src);
+  }
+
 public:
  /**
   * Reorder this MKLDNNMatrix from other format.