huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +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

2
candle-pyo3/Cargo.toml
View File

@ -18,6 +18,7 @@ candle = { path = "../candle-core", version = "0.3.0", package = "candle-core" }
candle-nn = { path = "../candle-nn", version = "0.3.0" }
half = { workspace = true }
pyo3 = { version = "0.19.0", features = ["extension-module"] }
intel-mkl-src = { workspace = true, optional = true }
[build-dependencies]
pyo3-build-config = "0.19"
@ -25,3 +26,4 @@ pyo3-build-config = "0.19"
[features]
default = []
cuda = ["candle/cuda"]
mkl = ["dep:intel-mkl-src","candle/mkl"]