diff options
Diffstat (limited to 'candle-nn/examples/basic_optimizer.rs')
| -rw-r--r-- | candle-nn/examples/basic_optimizer.rs | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/candle-nn/examples/basic_optimizer.rs b/candle-nn/examples/basic_optimizer.rs index cd5824dd..093bda81 100644 --- a/candle-nn/examples/basic_optimizer.rs +++ b/candle-nn/examples/basic_optimizer.rs @@ -1,3 +1,9 @@ +#[cfg(feature = "mkl")] +extern crate intel_mkl_src; + +#[cfg(feature = "accelerate")] +extern crate accelerate_src; + use candle::{DType, Device, Result, Tensor}; use candle_nn::{linear, AdamW, Linear, Module, ParamsAdamW, VarBuilder, VarMap}; |