fix hang on end of input without -v

zhanshen360 · Jan 23, 2017 · 7db5d5f · 7db5d5f
1 parent 660c20d
commit 7db5d5f
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/brainflayer.c b/brainflayer.c
@@ -845,15 +845,16 @@ int main(int argc, char **argv) {
             time_elapsed / 1.0e9
         );
 
-        if (batch_stopped < Bopt) {
-          fprintf(stderr, "\n");
-          break;
-        } else {
-          fflush(stderr);
-        }
+        fflush(stderr);
       }
     }
     // end stats
+
+    // main loop exit condition
+    if (batch_stopped < Bopt) {
+      if (vopt) { fprintf(stderr, "\n"); }
+      break;
+    }
   }
 
   return 0;