huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the missing kernel. (#955)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-24 17:21:37 +01:00
committed by GitHub
parent f5069dd354
commit 402ddcfcb4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-kernels/src/unary.cu
View File

@ -126,6 +126,7 @@ UNARY_OP1(__half, upowf_f16, powg(x, param))
UNARY_OP(uint8_t, ucopy_u8, x)
UNARY_OP(uint32_t, ucopy_u32, x)
UNARY_OP(int64_t, ucopy_i64, x)
UNARY_OP(float, ucopy_f32, x)
UNARY_OP(double, ucopy_f64, x)
UNARY_OP(float, uneg_f32, -x)