refactor: log util

thxCode · thxCode · commit 47a4656c2b66 · 2024-11-11T11:56:34.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/stable-diffusion.h b/stable-diffusion.h
@@ -109,6 +109,7 @@ SD_API void sd_log_set(sd_log_cb_t sd_log_cb, void* data);
 SD_API void sd_progress_set(sd_progress_cb_t cb, void* data);
 SD_API int32_t get_num_physical_cores();
 SD_API const char* sd_get_system_info();
+SD_API ggml_log_level sd_log_level_to_ggml_log_level(sd_log_level_t level);
 
 typedef struct {
     uint32_t width;
diff --git a/util.cpp b/util.cpp
@@ -423,6 +423,19 @@ const char* sd_get_system_info() {
     return buffer;
 }
 
+ggml_log_level sd_log_level_to_ggml_log_level(sd_log_level_t level) {
+    switch (level) {
+        case SD_LOG_INFO:
+            return GGML_LOG_LEVEL_INFO;
+        case SD_LOG_WARN:
+            return GGML_LOG_LEVEL_WARN;
+        case SD_LOG_ERROR:
+            return GGML_LOG_LEVEL_ERROR;
+        default:
+            return GGML_LOG_LEVEL_DEBUG;
+    }
+}
+
 sd_image_f32_t sd_image_t_to_sd_image_f32_t(sd_image_t image) {
     sd_image_f32_t converted_image;
     converted_image.width   = image.width;
