new comments

source/tensor/core/math/Compare.cuh

@@ -28,17 +28,11 @@ namespace nts{ // namespace nts(NiuTrans.Tensor)
 
 #ifdef USE_CUDA
 
-/* compare whether every entry is equal to the specified value (cuda kernel) */
-__global__
-void KernelEqual(DTYPE * a, DTYPE * b, DTYPE * number);
-/* compare whether every entry is equal to the specified value (cuda version) */
-void _CudaEqual(const XTensor * a, XTensor * b, DTYPE number);
-
-/* compare whether every entry is not equal to the specified value (cuda kernel) */
-__global__
-void KernelNotEqual(DTYPE * a, DTYPE * b, DTYPE * number);
-/* compare whether every entry is not equal to the specified value (cuda version) */
-void _CudaNotEqual(const XTensor * a, XTensor * b, DTYPE number);
+/* check whether every entry is equal to the given value (cuda version) */
+void _CudaEqual(const XTensor * a, XTensor * b, DTYPE value);
+
+/* check whether every entry is not equal to the given value (cuda version) */
+void _CudaNotEqual(const XTensor * a, XTensor * b, DTYPE value);
 
 #endif // USE_CUDA
 

source/tensor/core/math/Compare.h

@@ -26,23 +26,23 @@
 
 namespace nts{ // namespace nts(NiuTrans.Tensor)
 
-/* compare whether every entry is equal to the specified value */
-void _Equal(const XTensor * a, XTensor * b, DTYPE number);
-/* compare whether every entry is equal to the specified value (do it on site)
-keep the result in the input tensor a and return nothing */
-void _EqualMe(XTensor * a, DTYPE number);
-/* compare whether every entry is equal to the specified value (return an XTensor structure)
-make a new tensor to keep the result and return it */
-XTensor Equal(const XTensor & a, DTYPE number);
-
-/* compare whether every entry is not equal to the specified value */
-void _NotEqual(const XTensor * a, XTensor * b, DTYPE number);
-/* compare whether every entry is not equal to the specified value (do it on site)
-keep the result in the input tensor a and return nothing */
-void _NotEqualMe(XTensor * a, DTYPE number);
-/* compare whether every entry is not equal to the specified value (return an XTensor structure)
-make a new tensor to keep the result and return it */
-XTensor NotEqual(const XTensor & a, DTYPE number);
+/* check whether every entry is equal to the given value */
+void _Equal(const XTensor * a, XTensor * b, DTYPE value);
+
+/* check whether every entry is equal to the given value (do it on site) */
+void _EqualMe(XTensor * a, DTYPE value);
+
+/* check whether every entry is equal to the given value (return an XTensor structure) */
+XTensor Equal(const XTensor & a, DTYPE value);
+
+/* check whether every entry is not equal to the given value */
+void _NotEqual(const XTensor * a, XTensor * b, DTYPE value);
+
+/* check whether every entry is not equal to the given value (do it on site) */
+void _NotEqualMe(XTensor * a, DTYPE value);
+
+/* check whether every entry is not equal to the given value (return an XTensor structure) */
+XTensor NotEqual(const XTensor & a, DTYPE value);
 
 } // namespace nts(NiuTrans.Tensor)