re-add missing usqrt_bf16

huggingface · Jan 21, 2025 · eaf7eeb · eaf7eeb
1 parent a2f7148
commit eaf7eeb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/candle-kernels/src/unary.cu b/candle-kernels/src/unary.cu
@@ -112,6 +112,7 @@ UNARY_OP(__nv_bfloat16, uround_bf16, roundg(x))
 UNARY_OP(__nv_bfloat16, unormcdf_bf16, normcdfg(x))
 UNARY_OP(__nv_bfloat16, uabs_bf16, absg(x))
 UNARY_OP(__nv_bfloat16, usqr_bf16, x*x)
+UNARY_OP(__nv_bfloat16, usqrt_bf16, sqrtg(x))
 UNARY_OP(__nv_bfloat16, ugelu_bf16, gelu_fwd(x))
 UNARY_OP(__nv_bfloat16, ugelu_erf_bf16, gelu_erf_fwd(x))
 UNARY_OP(__nv_bfloat16, urelu_bf16, relu_fwd(x))