diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-06 11:05:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-06 11:05:05 +0100 |
| commit | c297a5096029edcc69117d3f1a7b97e7a5fc6767 (patch) | |
| tree | b3facbcedfdd8dfd88681a4930833fc25ffb8e95 /candle-core/examples | |
| parent | cd230d26fecb2ba69352a125d8ba1a4e75f3e6d1 (diff) | |
| download | candle-c297a5096029edcc69117d3f1a7b97e7a5fc6767.tar.gz candle-c297a5096029edcc69117d3f1a7b97e7a5fc6767.tar.bz2 candle-c297a5096029edcc69117d3f1a7b97e7a5fc6767.zip | |
Add mkl support for matrix multiply. (#86)
* Fix some rebase issues.
* Use mkl instead.
* Use mkl in bert.
* Add the optional mkl feature.
* Conditional compilation based on the mkl feature.
* Add more mkl support.
Diffstat (limited to 'candle-core/examples')
| -rw-r--r-- | candle-core/examples/basics.rs | 3 | ||||
| -rw-r--r-- | candle-core/examples/cuda_basics.rs | 3 |
2 files changed, 6 insertions, 0 deletions
diff --git a/candle-core/examples/basics.rs b/candle-core/examples/basics.rs index 733a7fe0..a5e2b24e 100644 --- a/candle-core/examples/basics.rs +++ b/candle-core/examples/basics.rs @@ -1,3 +1,6 @@ +#[cfg(feature = "mkl")] +extern crate intel_mkl_src; + use anyhow::Result; use candle::{Device, Tensor}; diff --git a/candle-core/examples/cuda_basics.rs b/candle-core/examples/cuda_basics.rs index c8852bf6..6050d793 100644 --- a/candle-core/examples/cuda_basics.rs +++ b/candle-core/examples/cuda_basics.rs @@ -1,3 +1,6 @@ +#[cfg(feature = "mkl")] +extern crate intel_mkl_src; + use anyhow::Result; use candle::{Device, Tensor}; |