Two more block-manager lints

Author: grahamking

lib/llm/src/block_manager/block/transfer/cuda.rs

@@ -225,8 +225,10 @@ unsafe fn cuda_memcpy_d2d(
         "Source and destination device memory regions must not overlap for D2D copy"
     );
 
-    cuda_result::memcpy_dtod_async(dst_ptr as u64, src_ptr as u64, size, stream.cu_stream())
-        .map_err(|e| TransferError::ExecutionError(format!("CUDA D2D memcpy failed: {}", e)))?;
+    unsafe {
+        cuda_result::memcpy_dtod_async(dst_ptr as u64, src_ptr as u64, size, stream.cu_stream())
+            .map_err(|e| TransferError::ExecutionError(format!("CUDA D2D memcpy failed: {}", e)))?
+    };
     Ok(())
 }
 

lib/llm/src/block_manager/connector/protocol.rs

@@ -194,12 +194,12 @@ impl TransferCompletionHandle for ScheduledTransferCompletionHandle {
     }
 
     async fn mark_complete(&self, result: anyhow::Result<()>) {
-        if let Some(completion_tx) = self.completion_tx.lock().unwrap().take() {
-            if completion_tx.send(result).is_err() {
-                tracing::error!(
-                    "failed to send completion status; this could lead to silent data corruption"
-                );
-            }
+        if let Some(completion_tx) = self.completion_tx.lock().unwrap().take()
+            && completion_tx.send(result).is_err()
+        {
+            tracing::error!(
+                "failed to send completion status; this could lead to silent data corruption"
+            );
         }
     }
 }