diff options
| -rw-r--r-- | Cargo.toml | 2 | ||||
| -rw-r--r-- | candle-pyo3/src/lib.rs | 3 |
2 files changed, 4 insertions, 1 deletions
@@ -3,7 +3,7 @@ members = [ "candle-core", "candle-examples", "candle-nn", - # "candle-pyo3", + "candle-pyo3", "candle-transformers", "candle-wasm-examples/llama2-c", "candle-wasm-examples/whisper", diff --git a/candle-pyo3/src/lib.rs b/candle-pyo3/src/lib.rs index 136f8a4f..c81cc713 100644 --- a/candle-pyo3/src/lib.rs +++ b/candle-pyo3/src/lib.rs @@ -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; |