From a93e8da7efaf6575facab1dba0ec8a903f80c5d4 Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Fri, 13 Dec 2024 14:30:35 -0500
Subject: [PATCH] Fix clippy

---
 crates/burn-fusion/src/tensor.rs | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

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<R: FusionRuntime> FusionTensor<R> {
         let id = self.stream;
         let client = self.client.clone();
         let desc = self.into_description();
-        let fut = client.read_tensor_float::<B>(desc, id);
-        async move { fut.await }
+        client.read_tensor_float::<B>(desc, id)
     }
 
     pub(crate) fn q_into_data<B>(self) -> impl Future<Output = TensorData>
@@ -136,8 +135,7 @@ impl<R: FusionRuntime> FusionTensor<R> {
             let id = self.stream;
             let client = self.client.clone();
             let desc = self.into_description();
-            let fut = client.read_tensor_quantized::<B>(desc, id);
-            async move { fut.await }
+            client.read_tensor_quantized::<B>(desc, id)
         } else {
             panic!("Expected quantized float dtype, got {:?}", self.dtype)
         }
@@ -150,7 +148,7 @@ impl<R: FusionRuntime> FusionTensor<R> {
         let id = self.stream;
         let client = self.client.clone();
         let desc = self.into_description();
-        async move { client.read_tensor_int::<B>(desc, id).await }
+        client.read_tensor_int::<B>(desc, id)
     }
 
     pub(crate) fn bool_into_data<B>(self) -> impl Future<Output = TensorData>
@@ -160,8 +158,7 @@ impl<R: FusionRuntime> FusionTensor<R> {
         let id = self.stream;
         let client = self.client.clone();
         let desc = self.into_description();
-        let fut = client.read_tensor_bool::<B>(desc, id);
-        async move { fut.await }
+        client.read_tensor_bool::<B>(desc, id)
     }
 }