Merge pull request torch#269 from dominikgrewe/maskedCopyFix

Fix loop logic error in maskedCopy.
shuzi · Jun 26, 2015 · 12ec0ad · 12ec0ad
2 parents 38bbb02 + 986ef6f
commit 12ec0ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/TH/generic/THTensorMath.c b/lib/TH/generic/THTensorMath.c
@@ -40,11 +40,11 @@ void THTensor_(maskedCopy)(THTensor *tensor, THByteTensor *mask, THTensor* src )
 		   }
 		   else if (*mask_data == 1)
 		   {
+		     if (cntr == nelem)
+		       THError("Number of elements of src < number of ones in mask");
 		     *tensor_data = *src_data;
 		     src_data++;
 		     cntr++;
-		     if (cntr > nelem)
-		       THError("Number of elements of src < number of ones in mask");
 		   });
   THTensor_(free)(srct);
 }