diff options
| author | Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> | 2023-10-23 21:10:59 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-23 20:10:59 +0100 |
| commit | eae94a451b3c6b3ef5975639e98dfbc587a2ac27 (patch) | |
| tree | 3e3241174fdbcdba145e388d2242b810363ee333 /candle-pyo3/src/lib.rs | |
| parent | 86e1803191e2ed44c57ad01807b29a886c0263bb (diff) | |
| download | candle-eae94a451b3c6b3ef5975639e98dfbc587a2ac27.tar.gz candle-eae94a451b3c6b3ef5975639e98dfbc587a2ac27.tar.bz2 candle-eae94a451b3c6b3ef5975639e98dfbc587a2ac27.zip | |
PyO3: Add `mkl` support (#1159)
* Add `mkl` support
* Set `mkl` path on linux
Diffstat (limited to 'candle-pyo3/src/lib.rs')
| -rw-r--r-- | candle-pyo3/src/lib.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-pyo3/src/lib.rs b/candle-pyo3/src/lib.rs index f16d8c1b..29f38ff8 100644 --- a/candle-pyo3/src/lib.rs +++ b/candle-pyo3/src/lib.rs @@ -8,6 +8,9 @@ use std::sync::Arc; use half::{bf16, f16}; +#[cfg(feature = "mkl")] +extern crate intel_mkl_src; + use ::candle::{quantized::QTensor, DType, Device, Tensor, WithDType}; pub fn wrap_err(err: ::candle::Error) -> PyErr { |