huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

PyO3 is back.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-25 22:44:30 +02:00
parent ed58de7551
commit b7814f66b4
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,6 +14,9 @@ pub fn wrap_err(err: ::candle::Error) -> PyErr {
#[pyclass(name = "Tensor")]
struct PyTensor(Tensor);
unsafe impl Send for PyTensor {}
unsafe impl Sync for PyTensor {}
impl std::ops::Deref for PyTensor {
type Target = Tensor;