commit 99f88f8833c020c2c8e88f2780f608ca0f86d958
author nicholas-leonard <nick@nikopia.org> Wed Jun 11 15:45:16 2014 -0400
committer nicholas-leonard <nick@nikopia.org> Wed Jun 11 15:45:16 2014 -0400
tree a5361efb5224c81f287644b8407cc9091027f17b
parent 7f9e557b038a777cc3482acb8f10a29d075c1af5

fixed argchecks

THCTensorMath.cu

diff --git a/THCTensorMath.cu b/THCTensorMath.cu
index 3d2375f..887d67a 100644
--- a/THCTensorMath.cu
+++ b/THCTensorMath.cu
@@ -1077,7 +1077,8 @@
   long size = THCudaTensor_nElement(data)/data->size[0];
   
   THArgCheck(dimension >= 0 && dimension < THCudaTensor_nDimension(src), 3, "invalid dimension");
-  THArgCheck(value > 0, 3, "non-positive-norm not supported");
+  THArgCheck(value > 0, 2, "non-positive-norm not supported");
+  THArgCheck(THCudaTensor_nDimension(src) > 1, 1, "need at least 2 dimensions");
 
   dim3 grid(data->size[0]);
   dim3 threads(32);