Add test_Matrix.cpp for auto compare member functions of class Matrix

paddle/math/tests/CMakeLists.txt

@@ -2,7 +2,7 @@
 
 add_simple_unittest(test_ExecViaCpu)
 add_simple_unittest(test_SIMDFunctions)
-add_simple_unittest(test_matrix)
+add_simple_unittest(test_SparseMatrix)
 
 # TODO(yuyang18): Refactor TestUtil.cpp. Remove this cross module reference.
 add_unittest(test_matrixCompare
@@ -14,5 +14,6 @@ add_simple_unittest(test_perturbation)
 add_simple_unittest(test_CpuGpuVector)
 add_simple_unittest(test_Allocator)
 add_simple_unittest(test_FPException)
-add_simple_unittest(test_BaseMatrix)
 add_simple_unittest(test_GpuProfiler)
+add_simple_unittest(test_BaseMatrix)
+add_simple_unittest(test_Matrix)

paddle/math/tests/TensorCheck.h

@@ -37,6 +37,7 @@ public:
         }
       }
     }
+
     return true;
   }
 
@@ -98,11 +99,15 @@ template<typename AssertEq, typename Tensor1, typename Tensor2>
 extern void TensorCheck(AssertEq compare,
                         const Tensor1& tensor1,
                         const Tensor2& tensor2) {
-  TensorCheck(
-    compare,
+  TensorCheck(compare,
               CopyToCpu<Tensor1>(tensor1).copiedArg(),
               CopyToCpu<Tensor2>(tensor2).copiedArg());
 }
 
-}  // namespace autotest
+template <typename AssertEq>
+void TensorCheck(AssertEq compare, real args1, real args2) {
+  EXPECT_EQ(compare(args1, args2), true) << "[Test error] args1 = " << args1
+                                         << ", args2 = " << args2;
+}
 
+}  // namespace autotest

paddle/math/tests/TestUtils.h

@@ -12,7 +12,6 @@ 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. */
 
-
 /**
  * TestUtils.h is used to automatically compare CPU and GPU code is consistent.
  *
@@ -51,83 +50,150 @@ public:
   typedef GpuMatrix type;
 };
 
+template <>
+class ReplaceType<Matrix, CpuMatrix> {
+public:
+  typedef CpuMatrix type;
+};
+
+template <>
+class ReplaceType<Matrix, GpuMatrix> {
+public:
+  typedef GpuMatrix type;
+};
+
 // construct a argument
-template<typename T> T construct(int height, int width);
-template<> float construct(int height, int width) { return 0.0; }
-template<> CpuMatrix construct(int height, int width) {
+template <typename T>
+T construct(int height, int width);
+template <>
+float construct(int height, int width) {
+  return 0.0;
+}
+template <>
+CpuMatrix construct(int height, int width) {
   CpuMatrix a(height, width);
   return a;
 }
-template<> GpuMatrix construct(int height, int width) {
+template <>
+GpuMatrix construct(int height, int width) {
   GpuMatrix a(height, width);
   return a;
 }
 
 // init a argument
-template<typename T> void init(T& v);
-template<> void init(float& v) { v = 0.5; }
-template<> void init(CpuMatrix& v) { v.randomizeUniform(); }
-template<> void init(GpuMatrix& v) { v.randomizeUniform(); }
+template <typename T>
+void init(T& v);
+template <>
+void init(float& v) {
+  v = 0.5;
+}
+template <>
+void init(CpuMatrix& v) {
+  v.randomizeUniform();
+}
+template <>
+void init(GpuMatrix& v) {
+  v.randomizeUniform();
+}
 
 // init a tuple which contains a set of arguments.
 template <std::size_t I = 0, typename... Args>
-inline typename std::enable_if<I == sizeof...(Args), void>::type
-initTuple(std::tuple<Args...>& t){}
+inline typename std::enable_if<I == sizeof...(Args), void>::type initTuple(
+    std::tuple<Args...>& t) {}
 
 template <std::size_t I = 0, typename... Args>
-inline typename std::enable_if<I < sizeof...(Args), void>::type
-initTuple(std::tuple<Args...>& t) {
+    inline typename std::enable_if <
+    I<sizeof...(Args), void>::type initTuple(std::tuple<Args...>& t) {
   init(std::get<I>(t));
   initTuple<I + 1>(t);
 }
 
 // copy a argument, copy src to dest
-template<typename T1, typename T2> void copy(T1& dest, T2& src);
-template<> void copy(float& dest, float& src) { dest = src; }
-template<> void copy(GpuMatrix& dest, CpuMatrix& src) {
+template <typename T1, typename T2>
+void copy(T1& dest, T2& src);
+template <>
+void copy(float& dest, float& src) {
+  dest = src;
+}
+template <>
+void copy(GpuMatrix& dest, CpuMatrix& src) {
   dest.copyFrom(src);
 }
 
 // copy a tuple, copy src to dest
 template <std::size_t I = 0, typename... Args1, typename... Args2>
-inline typename std::enable_if<I == sizeof...(Args1), void>::type
-copyTuple(std::tuple<Args1...>& dest, std::tuple<Args2...>& src) {}
+inline typename std::enable_if<I == sizeof...(Args1), void>::type copyTuple(
+    std::tuple<Args1...>& dest, std::tuple<Args2...>& src) {}
 
 template <std::size_t I = 0, typename... Args1, typename... Args2>
-inline typename std::enable_if<I < sizeof...(Args1), void>::type
-copyTuple(std::tuple<Args1...>& dest, std::tuple<Args2...>& src) {
+    inline typename std::enable_if <
+    I<sizeof...(Args1), void>::type copyTuple(std::tuple<Args1...>& dest,
+                                              std::tuple<Args2...>& src) {
   copy(std::get<I>(dest), std::get<I>(src));
   copyTuple<I + 1>(dest, src);
 }
 
+// Compare output
+template <std::size_t I = 0,
+          typename... Args1,
+          typename... Args2,
+          typename AssertEq>
+inline typename std::enable_if<I == sizeof...(Args1), void>::type checkTuple(
+    std::tuple<Args1...>& args1,
+    std::tuple<Args2...>& args2,
+    AssertEq compare) {}
+
+template <std::size_t I = 0,
+          typename... Args1,
+          typename... Args2,
+          typename AssertEq>
+    inline typename std::enable_if <
+    I<sizeof...(Args1), void>::type checkTuple(std::tuple<Args1...>& args1,
+                                               std::tuple<Args2...>& args2,
+                                               AssertEq compare) {
+  TensorCheck(compare, std::get<I>(args1), std::get<I>(args2));
+  checkTuple<I + 1>(args1, args2, compare);
+}
+
 // call member function
 template <typename C,
-          typename FC, typename R, typename ...FArgs, typename ...Args>
+          typename FC,
+          typename R,
+          typename... FArgs,
+          typename... Args>
 R call(C& obj, R (FC::*f)(FArgs...), Args&&... args) {
   return (obj.*f)(args...);
 }
 
-template <bool ApplyRow, bool ApplyCol,
-          std::size_t... I, typename C, typename R, typename ...Args,
+template <bool ApplyRow,
+          bool ApplyCol,
+          std::size_t... I,
+          typename C,
+          typename R,
+          typename... Args,
           typename AssertEq>
-void BaseMatrixCompare(R (C::*f)(Args...), AssertEq compare) {
+void BaseMatrixCompare(R (C::*f)(Args...),
+                       AssertEq compare,
+                       bool checkArgs = false) {
   for (auto height : {1, 11, 73, 128, 200, 330}) {
     for (auto width : {1, 3, 32, 100, 512, 1000}) {
-      CpuMatrix obj1(ApplyCol ? 1 : height,
-                     ApplyRow ? 1 : width);
-      GpuMatrix obj2(ApplyCol ? 1 : height,
-                     ApplyRow ? 1 : width);
+      CpuMatrix obj1(ApplyCol ? 1 : height, ApplyRow ? 1 : width);
+      GpuMatrix obj2(ApplyCol ? 1 : height, ApplyRow ? 1 : width);
       init(obj1);
       copy(obj2, obj1);
 
       auto tuple1 = std::make_tuple(
-        construct<typename ReplaceType<typename std::decay<
-          typename std::tuple_element<I, std::tuple<Args...>>::type>::type,
+          construct<typename ReplaceType<
+              typename std::decay<
+                  typename std::tuple_element<I,
+                                              std::tuple<Args...>>::type>::type,
               CpuMatrix>::type>(height, width)...);
 
       auto tuple2 = std::make_tuple(
-        construct<typename ReplaceType<typename std::decay<
-          typename std::tuple_element<I, std::tuple<Args...>>::type>::type,
+          construct<typename ReplaceType<
+              typename std::decay<
+                  typename std::tuple_element<I,
+                                              std::tuple<Args...>>::type>::type,
               GpuMatrix>::type>(height, width)...);
 
       initTuple(tuple1);
@@ -137,6 +203,9 @@ void BaseMatrixCompare(R (C::*f)(Args...), AssertEq compare) {
       call(obj2, f, std::get<I>(tuple2)...);
 
       TensorCheck(compare, obj1, obj2);
+      if (checkArgs) {
+        checkTuple(tuple1, tuple2, compare);
+      }
     }
   }
 }
@@ -144,7 +213,7 @@ void BaseMatrixCompare(R (C::*f)(Args...), AssertEq compare) {
 }  // namespace autotest
 
 template <std::size_t... I, typename C, typename R, typename... Args>
-void BaseMatrixCompare(R (C::*f)(Args...)) {
+void BaseMatrixCompare(R (C::*f)(Args...), bool checkArgs = false) {
   static_assert(sizeof...(I) == sizeof...(Args),
                 "size of parameter packs are not equal");
 
@@ -154,7 +223,7 @@ void BaseMatrixCompare(R (C::*f)(Args...)) {
   autotest::AssertEqual compare(1e-10);
 #endif
 
-  autotest::BaseMatrixCompare<false, false, I...>(f, compare);
+  autotest::BaseMatrixCompare<false, false, I...>(f, compare, checkArgs);
 }
 
 template <std::size_t... I, typename C, typename R, typename... Args>
@@ -183,4 +252,3 @@ void BaseMatrixApplyCol(R (C::*f)(Args...)) {
 #endif
   autotest::BaseMatrixCompare<false, true, I...>(f, compare);
 }
-

paddle/math/tests/test_Matrix.cpp

@@ -0,0 +1,49 @@
+/* Copyright (c) 2016 Baidu, Inc. All Rights Reserve.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+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. */
+
+#ifndef PADDLE_ONLY_CPU
+/**
+ * This test file compares the implementation of CPU and GPU function
+ * in Matrix.cpp.
+ */
+
+#include <gtest/gtest.h>
+#include "paddle/utils/Util.h"
+#include "paddle/math/BaseMatrix.h"
+#include "TestUtils.h"
+
+using namespace paddle;  // NOLINT
+using namespace std;     // NOLINT
+
+/**
+ * Test member functions which prototype is
+ * void (Matrix::*)(Matrix&).
+ */
+TEST(BaseMatrix, real) {
+  typedef void (Matrix::*FunctionProto)(Matrix&);
+#define MATRIXCOMPARE(function) \
+  BaseMatrixCompare<0>(static_cast<FunctionProto>(&Matrix::function), true);
+
+  MATRIXCOMPARE(softmax);
+
+#undef MATRIXCOMPARE
+}
+
+int main(int argc, char** argv) {
+  testing::InitGoogleTest(&argc, argv);
+  initMain(argc, argv);
+  return RUN_ALL_TESTS();
+}
+
+#endif

paddle/math/tests/test_matrixCompare.cpp

@@ -693,24 +693,6 @@ TEST(Matrix, unary) {
   }
 }
 
-void testMatrixSoftmax(int height, int width) {
-  MatrixPtr cpuInput = std::make_shared<CpuMatrix>(height, width);
-  MatrixPtr cpuOutput = std::make_shared<CpuMatrix>(height, width);
-  MatrixPtr gpuInput = std::make_shared<GpuMatrix>(height, width);
-  MatrixPtr gpuOutput = std::make_shared<GpuMatrix>(height, width);
-
-  cpuInput->randomizeUniform();
-  gpuInput->copyFrom(*cpuInput);
-  cpuOutput->zero();
-  gpuOutput->zero();
-  cpuInput->softmax(*cpuOutput);
-  gpuInput->softmax(*gpuOutput);
-
-  MatrixPtr outputCheck = std::make_shared<CpuMatrix>(height, width);
-  outputCheck->copyFrom(*gpuOutput);
-  MatrixCheckErr(*cpuOutput, *outputCheck);
-}
-
 void testSequenceSoftmax(int batchSize) {
   // forward
   int inputDim = 1;
@@ -793,7 +775,6 @@ TEST(Matrix, softmax) {
     for (auto width : {1, 32, 100, 512, 1000}) {
       VLOG(3) << " height=" << height << " width=" << width;
 
-      testMatrixSoftmax(height, width);
       testMatrixSoftmaxBp(height, width);
       testMatrixSoftmaxThreshold(height, width);
     }