diff options
author | soumith <soumith@fb.com> | 2014-11-06 20:50:19 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2014-11-06 20:50:19 +0300 |
commit | e87521e17d0fd9b39fd03d857203f103a6fbdeb9 (patch) | |
tree | 5137e0fc497a205a193fd678d2f6acd22b096259 /TensorOperator.c | |
parent | 701cc5e20c289a94dc5ac077a2216998ace8cebe (diff) |
fixing cadd API inconsistency
Diffstat (limited to 'TensorOperator.c')
-rw-r--r-- | TensorOperator.c | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/TensorOperator.c b/TensorOperator.c index bd30465..d2c77e3 100644 --- a/TensorOperator.c +++ b/TensorOperator.c @@ -30,7 +30,7 @@ static int cutorch_CudaTensorOperator___add__(lua_State *L) { THCudaTensor_resizeAs(r, tensor1); THCudaTensor_copy(r, tensor1); - THCudaTensor_cadd_tst(r, r, 1, tensor2); + THCudaTensor_cadd(r, r, 1, tensor2); } } return 1; @@ -53,7 +53,7 @@ static int cutorch_CudaTensorOperator___sub__(lua_State *L) { THCudaTensor_resizeAs(r, tensor2); THCudaTensor_fill(r, luaL_checknumber(L, 1)); - THCudaTensor_cadd_tst(r, r, -1, tensor2); + THCudaTensor_cadd(r, r, -1, tensor2); } else if(tensor1 && !tensor2) { @@ -65,7 +65,7 @@ static int cutorch_CudaTensorOperator___sub__(lua_State *L) { THCudaTensor_resizeAs(r, tensor1); THCudaTensor_copy(r, tensor1); - THCudaTensor_cadd_tst(r, r, -1, tensor2); + THCudaTensor_cadd(r, r, -1, tensor2); } } return 1; |