diff --git a/crates/burn-fusion/src/tensor.rs b/crates/burn-fusion/src/tensor.rs index 22eb22c490..f620e2c722 100644 --- a/crates/burn-fusion/src/tensor.rs +++ b/crates/burn-fusion/src/tensor.rs @@ -124,8 +124,7 @@ impl FusionTensor { let id = self.stream; let client = self.client.clone(); let desc = self.into_description(); - let fut = client.read_tensor_float::(desc, id); - async move { fut.await } + client.read_tensor_float::(desc, id) } pub(crate) fn q_into_data(self) -> impl Future @@ -136,8 +135,7 @@ impl FusionTensor { let id = self.stream; let client = self.client.clone(); let desc = self.into_description(); - let fut = client.read_tensor_quantized::(desc, id); - async move { fut.await } + client.read_tensor_quantized::(desc, id) } else { panic!("Expected quantized float dtype, got {:?}", self.dtype) } @@ -150,7 +148,7 @@ impl FusionTensor { let id = self.stream; let client = self.client.clone(); let desc = self.into_description(); - async move { client.read_tensor_int::(desc, id).await } + client.read_tensor_int::(desc, id) } pub(crate) fn bool_into_data(self) -> impl Future @@ -160,8 +158,7 @@ impl FusionTensor { let id = self.stream; let client = self.client.clone(); let desc = self.into_description(); - let fut = client.read_tensor_bool::(desc, id); - async move { fut.await } + client.read_tensor_bool::(desc, id) } }