commit 3ad4f6e3f37b29530d5d2572c2f00df42906aaee
author nicholas-leonard <nick@nikopia.org> Sat Jul 19 18:32:42 2014 -0400
committer nicholas-leonard <nick@nikopia.org> Sat Jul 19 18:32:42 2014 -0400
tree 879e1686c56ee8820b340814f642a7f789450a53
parent d93a56dc3651995b9ef3c3b2ca145ac5e8dfe791

added back arg checks

THCTensorMath.cu

diff --git a/THCTensorMath.cu b/THCTensorMath.cu
index f031f8c..62ebae1 100644
--- a/THCTensorMath.cu
+++ b/THCTensorMath.cu
@@ -89,6 +89,7 @@
 
 void THCudaTensor_cadd_tst(THCudaTensor *self_, THCudaTensor* src1, float value, THCudaTensor *src2)
 {
+  THArgCheck(THCudaTensor_nElement(src1) == THCudaTensor_nElement(src2), 3, "size do not match");
   THCudaTensor_resizeAs(self_, src1);
   {
     THCudaTensor *self = THCudaTensor_newContiguous(self_);
@@ -108,6 +109,7 @@
 
 void THCudaTensor_cmul(THCudaTensor *self_, THCudaTensor *src1, THCudaTensor *src2)
 {
+  THArgCheck(THCudaTensor_nElement(src1) == THCudaTensor_nElement(src2), 3, "size do not match");
   THCudaTensor_resizeAs(self_, src1);
   {
     THCudaTensor *self = THCudaTensor_newContiguous(self_);
@@ -128,7 +130,7 @@
 
 void THCudaTensor_cdiv(THCudaTensor *self_, THCudaTensor *src)
 {
-  THCudaTensor_resizeAs(self_, src);
+  THArgCheck(THCudaTensor_nElement(self_) == THCudaTensor_nElement(src), 2, "size do not match");
   {
     THCudaTensor *self = THCudaTensor_newContiguous(self_);
     long size = THCudaTensor_nElement(self);
@@ -154,6 +156,7 @@
 
 void THCudaTensor_addcmul(THCudaTensor *self_, float value, THCudaTensor *src1, THCudaTensor *src2)
 {
+  THArgCheck(THCudaTensor_nElement(src1) == THCudaTensor_nElement(src2), 3, "size do not match");
   THCudaTensor_resizeAs(self_, src1);
   {
     THCudaTensor *self = THCudaTensor_newContiguous(self_);
@@ -189,6 +192,7 @@
 
 void THCudaTensor_addcdiv(THCudaTensor *self_, float value, THCudaTensor *src1, THCudaTensor *src2)
 {
+  THArgCheck(THCudaTensor_nElement(src1) == THCudaTensor_nElement(src2), 3, "size do not match");
   THCudaTensor_resizeAs(self_, src1);
   {
     THCudaTensor *self = THCudaTensor_newContiguous(self_);
@@ -928,6 +932,7 @@
 template<class Op>
 void THCudaTensor_logicalTensor(THCudaTensor *self_, THCudaTensor *src1, THCudaTensor *src2, Op op)
 {
+  THArgCheck(THCudaTensor_nElement(src1) == THCudaTensor_nElement(src2), 3, "size do not match");
   THCudaTensor_resizeAs(self_, src1);
 
   THCudaTensor *self = THCudaTensor_newContiguous(self_);