fix: flushes after printf (leejet#38)

dmikey · web-flow · commit 1b5a86829620 · 2023-08-30T20:47:25.000+08:00
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -26,12 +26,16 @@ static SDLogLevel log_level = SDLogLevel::INFO;
         }                                                                                             \
         if (level == SDLogLevel::DEBUG) {                                                             \
             printf("[DEBUG] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__);          \
+            fflush(stdout);                                                                           \
         } else if (level == SDLogLevel::INFO) {                                                       \
             printf("[INFO]  %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__);          \
+            fflush(stdout);                                                                           \
         } else if (level == SDLogLevel::WARN) {                                                       \
             fprintf(stderr, "[WARN]  %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \
+            fflush(stdout);                                                                           \
         } else if (level == SDLogLevel::ERROR) {                                                      \
             fprintf(stderr, "[ERROR] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \
+            fflush(stdout);                                                                           \
         }                                                                                             \
     } while (0)
 
@@ -135,6 +139,7 @@ float ggml_tensor_get_f32(const ggml_tensor* tensor, int l, int k = 0, int j = 0
 
 void print_ggml_tensor(struct ggml_tensor* tensor, bool shape_only = false) {
     printf("shape(%zu, %zu, %zu, %zu)\n", tensor->ne[0], tensor->ne[1], tensor->ne[2], tensor->ne[3]);
+    fflush(stdout);
     if (shape_only) {
         return;
     }
@@ -156,6 +161,7 @@ void print_ggml_tensor(struct ggml_tensor* tensor, bool shape_only = false) {
                         continue;
                     }
                     printf("  [%d, %d, %d, %d] = %f\n", i, j, k, l, ggml_tensor_get_f32(tensor, l, k, j, i));
+                    fflush(stdout);
                 }
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -26,12 +26,16 @@ static SDLogLevel log_level = SDLogLevel::INFO;`
`26`	`26`	`} \`
`27`	`27`	`if (level == SDLogLevel::DEBUG) { \`
`28`	`28`	`printf("[DEBUG] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \`
	`29`	`+ fflush(stdout); \`
`29`	`30`	`} else if (level == SDLogLevel::INFO) { \`
`30`	`31`	`printf("[INFO] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \`
	`32`	`+ fflush(stdout); \`
`31`	`33`	`} else if (level == SDLogLevel::WARN) { \`
`32`	`34`	`fprintf(stderr, "[WARN] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \`
	`35`	`+ fflush(stdout); \`
`33`	`36`	`} else if (level == SDLogLevel::ERROR) { \`
`34`	`37`	`fprintf(stderr, "[ERROR] %s:%-4d - " format "\n", __FILENAME__, __LINE__, ##__VA_ARGS__); \`
	`38`	`+ fflush(stdout); \`
`35`	`39`	`} \`
`36`	`40`	`} while (0)`
`37`	`41`
`@@ -135,6 +139,7 @@ float ggml_tensor_get_f32(const ggml_tensor* tensor, int l, int k = 0, int j = 0`
`135`	`139`
`136`	`140`	`void print_ggml_tensor(struct ggml_tensor* tensor, bool shape_only = false) {`
`137`	`141`	`printf("shape(%zu, %zu, %zu, %zu)\n", tensor->ne[0], tensor->ne[1], tensor->ne[2], tensor->ne[3]);`
	`142`	`+ fflush(stdout);`
`138`	`143`	`if (shape_only) {`
`139`	`144`	`return;`
`140`	`145`	`}`
`@@ -156,6 +161,7 @@ void print_ggml_tensor(struct ggml_tensor* tensor, bool shape_only = false) {`
`156`	`161`	`continue;`
`157`	`162`	`}`
`158`	`163`	`printf(" [%d, %d, %d, %d] = %f\n", i, j, k, l, ggml_tensor_get_f32(tensor, l, k, j, i));`
	`164`	`+ fflush(stdout);`
`159`	`165`	`}`
`160`	`166`	`}`
`161`	`167`	`}`