diff options
| author | Gonzalo <456459+grzuy@users.noreply.github.com> | 2023-09-21 15:52:39 -0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-21 19:52:39 +0100 |
| commit | a96878f2357fbcebf9db8747dcbb55bc8200d8ab (patch) | |
| tree | 0d0df79ef039bc982efc9896a083525dcb155966 /candle-kernels | |
| parent | aa8ec06fd2b02c1039a46fcb518fd6d351487978 (diff) | |
| download | candle-a96878f2357fbcebf9db8747dcbb55bc8200d8ab.tar.gz candle-a96878f2357fbcebf9db8747dcbb55bc8200d8ab.tar.bz2 candle-a96878f2357fbcebf9db8747dcbb55bc8200d8ab.zip | |
cuda cast i64 (#925)
Diffstat (limited to 'candle-kernels')
| -rw-r--r-- | candle-kernels/src/cast.cu | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/candle-kernels/src/cast.cu b/candle-kernels/src/cast.cu index ab2045a3..ee20fe5f 100644 --- a/candle-kernels/src/cast.cu +++ b/candle-kernels/src/cast.cu @@ -77,20 +77,30 @@ CAST_OP(double, __half, cast_f64_f16) CAST_OP(uint32_t, uint32_t, cast_u32_u32) CAST_OP(uint32_t, uint8_t, cast_u32_u8 ) +CAST_OP(uint32_t, int64_t, cast_u32_i64 ) CAST_OP(uint32_t, float, cast_u32_f32) CAST_OP(uint32_t, double, cast_u32_f64) CAST_OP(uint8_t, uint32_t, cast_u8_u32) CAST_OP(uint8_t, uint8_t, cast_u8_u8 ) +CAST_OP(uint8_t, int64_t, cast_u8_i64 ) CAST_OP(uint8_t, float, cast_u8_f32) CAST_OP(uint8_t, double, cast_u8_f64) +CAST_OP(int64_t, uint32_t, cast_i64_u32) +CAST_OP(int64_t, uint8_t, cast_i64_u8 ) +CAST_OP(int64_t, int64_t, cast_i64_i64 ) +CAST_OP(int64_t, float, cast_i64_f32) +CAST_OP(int64_t, double, cast_i64_f64) + CAST_OP(float, uint8_t, cast_f32_u8 ) CAST_OP(float, uint32_t, cast_f32_u32) +CAST_OP(float, int64_t, cast_f32_i64 ) CAST_OP(float, float, cast_f32_f32) CAST_OP(float, double, cast_f32_f64) CAST_OP(double, uint8_t, cast_f64_u8 ) CAST_OP(double, uint32_t, cast_f64_u32) +CAST_OP(double, int64_t, cast_f64_i64 ) CAST_OP(double, float, cast_f64_f32) CAST_OP(double, double, cast_f64_f64) |