huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

PyO3: Add mkl support (#1159)

Browse Source 
* Add `mkl` support

* Set `mkl` path on linux
This commit is contained in:
Lukas Kreussel
2023-10-23 21:10:59 +02:00
committed by GitHub
parent 86e1803191
commit eae94a451b
3 changed files with 42 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-pyo3/src/lib.rs
View File

@ -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 {