diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-08 21:58:56 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-08 21:58:56 +0100 |
| commit | 057f7909bc5544b0dcb0e6d5360288633dbf443f (patch) | |
| tree | e6aaea33d24479fe819d3a13bf6a7a4747cd8025 /candle-core/src/op.rs | |
| parent | acf8f10ae17d7f472dc1a634fbd7358a79d7b4d4 (diff) | |
| download | candle-057f7909bc5544b0dcb0e6d5360288633dbf443f.tar.gz candle-057f7909bc5544b0dcb0e6d5360288633dbf443f.tar.bz2 candle-057f7909bc5544b0dcb0e6d5360288633dbf443f.zip | |
Accelerate support for gelu. (#782)
Diffstat (limited to 'candle-core/src/op.rs')
| -rw-r--r-- | candle-core/src/op.rs | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/candle-core/src/op.rs b/candle-core/src/op.rs index fbfc9c1a..9382b217 100644 --- a/candle-core/src/op.rs +++ b/candle-core/src/op.rs @@ -600,6 +600,24 @@ impl UnaryOpT for Gelu { fn f64_vec(xs: &[f64], ys: &mut [f64]) { crate::mkl::vd_gelu(xs, ys) } + + #[cfg(feature = "accelerate")] + const F32_VEC: bool = true; + + #[cfg(feature = "accelerate")] + #[inline(always)] + fn f32_vec(xs: &[f32], ys: &mut [f32]) { + crate::accelerate::vs_gelu(xs, ys) + } + + #[cfg(feature = "accelerate")] + const F64_VEC: bool = true; + + #[cfg(feature = "accelerate")] + #[inline(always)] + fn f64_vec(xs: &[f64], ys: &mut [f64]) { + crate::accelerate::vd_gelu(xs, ys) + } } impl UnaryOpT for Relu { |