Skip to content

Commit 3169921

Browse files
author
“thucydides”
committed
fix: Use manual copy_data in tests to avoid trait issues
1 parent f8171e1 commit 3169921

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

rustorch-pytorch/src/lib.rs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,9 @@ mod tests {
122122
// Convert to PyTorch
123123
let torch_tensor = PyTorchAdapter::to_torch(&rt_tensor);
124124
assert_eq!(torch_tensor.size(), vec![2, 2]);
125-
let data_vec: Vec<f32> = torch_tensor.shallow_clone().into();
125+
let numel = torch_tensor.numel();
126+
let mut data_vec = vec![0.0f32; numel as usize];
127+
torch_tensor.copy_data(&mut data_vec, numel as usize);
126128
assert_eq!(data_vec, data);
127129

128130
// Convert back

0 commit comments

Comments
 (0)