huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for accelerate in the pyo3 bindings. (#1167)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-24 06:34:37 +01:00
committed by GitHub
parent 807e3f9f52
commit 7bd0faba75
3 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-pyo3/test.py
View File

@ -1,5 +1,10 @@
import candle
print(f"mkl: {candle.utils.has_mkl()}")
print(f"accelerate: {candle.utils.has_accelerate()}")
print(f"num-threads: {candle.utils.get_num_threads()}")
print(f"cuda: {candle.utils.cuda_is_available()}")
t = candle.Tensor(42.0)
print(t)
print(t.shape, t.rank, t.device)