break when there's nothing to read

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
huggingface · Sep 28, 2024 · 6d95155 · 6d95155
1 parent 5b6b74e
commit 6d95155
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/launcher/src/main.rs b/launcher/src/main.rs
@@ -922,6 +922,8 @@ fn shard_manager(
             if let Ok(n) = stdin.read(&mut buffer) {
                 if n > 0 {
                     let _ = pstdin.write_all(&buffer[..n]);
+                } else {
+                    break;
                 }
             }
         }
@@ -1063,6 +1065,8 @@ fn log_lines<R: Sized + Read>(mut bufread: BufReader<R>) {
                         }
                     }
                 }
+            } else {
+                break;
             }
         }
     }