Fix clippy

tracel-ai · Dec 13, 2024 · a93e8da · a93e8da
1 parent ba03fad
commit a93e8da
Showing 1 changed file with 4 additions and 7 deletions.
diff --git 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)
     }
 }