diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-07-25 22:53:09 +0200 |
|---|---|---|
| committer | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-07-27 09:58:47 +0200 |
| commit | 1553b58fe59a29fe808b9b4d43a6502046ce26dd (patch) | |
| tree | d541e5603eca7579b0e9d449837db68c8813b3fe | |
| parent | b7814f66b4844f2f0c5d35aed465f4f40a5f6845 (diff) | |
| download | candle-1553b58fe59a29fe808b9b4d43a6502046ce26dd.tar.gz candle-1553b58fe59a29fe808b9b4d43a6502046ce26dd.tar.bz2 candle-1553b58fe59a29fe808b9b4d43a6502046ce26dd.zip | |
Tensor are not necessarily sendable (CustomOp1).
| -rw-r--r-- | candle-pyo3/src/lib.rs | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/candle-pyo3/src/lib.rs b/candle-pyo3/src/lib.rs index c81cc713..6e206688 100644 --- a/candle-pyo3/src/lib.rs +++ b/candle-pyo3/src/lib.rs @@ -11,12 +11,9 @@ pub fn wrap_err(err: ::candle::Error) -> PyErr { } #[derive(Clone)] -#[pyclass(name = "Tensor")] +#[pyclass(name = "Tensor", unsendable)] struct PyTensor(Tensor); -unsafe impl Send for PyTensor {} -unsafe impl Sync for PyTensor {} - impl std::ops::Deref for PyTensor { type Target = Tensor; |