Bug fixed.

Minor bugs fixed.

source/tensor/core/arithmetic/Mask.cu

@@ -22,6 +22,7 @@
 */
 
 #include "../../XDevice.h"
+#include "../../XTensor.h"
 #include "../../XUtility.h"
 
 namespace nts { // namespace nts(NiuTrans.Tensor)

source/tensor/test/TSum.cpp

@@ -286,7 +286,7 @@ bool TestSum3()
     cUserGPU = Sum(*aGPU, *bGPU, beta);
 
     /* check results */
-    gpuTest = _CheckData(&cUserGPU, answer, cUunitNum);
+    gpuTest = _CheckData(&cUserGPU, answer, cUnitNum);
 
     /* destroy variables */
     delete a;
@@ -388,7 +388,7 @@ bool TestSum4()
     cUserGPU = Sum(*aGPU, *bGPU, beta);
 
     /* check results */
-    gpuTest = _CheckData(&cUserGPU, answer, cUunitNum);
+    gpuTest = _CheckData(&cUserGPU, answer, cUnitNum);
 
     /* destroy variables */
     delete a;
@@ -490,7 +490,7 @@ bool TestSum5()
     cUserGPU = Sum(*aGPU, *bGPU, beta);
 
     /* check results */
-    gpuTest = _CheckData(&cUserGPU, answer, cUunitNum);
+    gpuTest = _CheckData(&cUserGPU, answer, cUnitNum);
 
     /* destroy variables */
     delete a;