Added support for neg().

zhouyonglong · Nov 12, 2015 · a8c49df · a8c49df
1 parent b9b4ae6
commit a8c49df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/THCTensorMath.h b/THCTensorMath.h
@@ -70,6 +70,7 @@ THC_API void THCudaTensor_sqrt(THCState *state, THCudaTensor *self, THCudaTensor
 THC_API void THCudaTensor_ceil(THCState *state, THCudaTensor *self, THCudaTensor *src);
 THC_API void THCudaTensor_floor(THCState *state, THCudaTensor *self, THCudaTensor *src);
 THC_API void THCudaTensor_abs(THCState *state, THCudaTensor *self, THCudaTensor *src);
+THC_API void THCudaTensor_neg(THCState *state, THCudaTensor *self, THCudaTensor *src);
 THC_API void THCudaTensor_sign(THCState *state, THCudaTensor *self, THCudaTensor *src);
 THC_API void THCudaTensor_round(THCState *state, THCudaTensor *self, THCudaTensor *src);
 THC_API void THCudaTensor_atan2(THCState *state, THCudaTensor *r_, THCudaTensor *tx, THCudaTensor *ty);

diff --git a/THCTensorMathPointwise.cu b/THCTensorMathPointwise.cu
@@ -50,6 +50,7 @@ IMPLEMENT_CUDA_TENSOR_BASIC_FUNC(ceil, ceil)
 IMPLEMENT_CUDA_TENSOR_BASIC_FUNC(floor, floor)
 IMPLEMENT_CUDA_TENSOR_BASIC_FUNC(abs, fabs)
 IMPLEMENT_CUDA_TENSOR_BASIC_FUNC(round, roundf)
+IMPLEMENT_CUDA_TENSOR_BASIC_FUNC(neg, -)
 
 #undef IMPLEMENT_CUDA_TENSOR_BASIC_FUNC