diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-07 06:42:57 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-07 06:42:57 +0100 |
| commit | f114394456c5b569f21558e53bca86fe1ecd9c76 (patch) | |
| tree | b636eabc23357ba4e152a623e2e00b0a74e9d73b /candle-kernels/src | |
| parent | bac4ef40f328fb11d99d6653ac4b2bd5c743ea11 (diff) | |
| download | candle-f114394456c5b569f21558e53bca86fe1ecd9c76.tar.gz candle-f114394456c5b569f21558e53bca86fe1ecd9c76.tar.bz2 candle-f114394456c5b569f21558e53bca86fe1ecd9c76.zip | |
Include the math.h file to get access to constants. (#99)
Diffstat (limited to 'candle-kernels/src')
| -rw-r--r-- | candle-kernels/src/unary.cu | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-kernels/src/unary.cu b/candle-kernels/src/unary.cu index 726db339..bd8a210e 100644 --- a/candle-kernels/src/unary.cu +++ b/candle-kernels/src/unary.cu @@ -1,4 +1,6 @@ #include "cuda_utils.cuh" +#define _USE_MATH_DEFINES +#include<math.h> #include<stdint.h> #define UNARY_OP(TYPENAME, FN_NAME, FUNC) \ |