From 9aca398a4fcd87aa095e933583190926359094bf Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sun, 13 Aug 2023 13:53:34 +0200 Subject: More accelerate optimizations (#427) * Add more tracing to the whisper example. * Support accelerate in more examples. * Use accelerate for pointwise functions. * Use accelerate for binary operations too. * Bugfix for binary operation: use the rhs before the lhs. --- candle-core/examples/cuda_sum_benchmark.rs | 3 +++ 1 file changed, 3 insertions(+) (limited to 'candle-core/examples/cuda_sum_benchmark.rs') diff --git a/candle-core/examples/cuda_sum_benchmark.rs b/candle-core/examples/cuda_sum_benchmark.rs index 1c8b0136..d6d182e8 100644 --- a/candle-core/examples/cuda_sum_benchmark.rs +++ b/candle-core/examples/cuda_sum_benchmark.rs @@ -1,6 +1,9 @@ #[cfg(feature = "mkl")] extern crate intel_mkl_src; +#[cfg(feature = "accelerate")] +extern crate accelerate_src; + use std::str::FromStr; use anyhow::Result; -- cgit v1.2.3