Skip to content

Commit 94b4f06

Browse files
tinglouarthw
authored andcommitted
llava: return false instead of exit (ggml-org#10546)
1 parent 1cb83e0 commit 94b4f06

File tree

2 files changed

+30
-13
lines changed

2 files changed

+30
-13
lines changed

examples/llava/clip.cpp

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,10 +40,17 @@
4040
#include <cinttypes>
4141
#include <limits>
4242

43-
#define LOG_INF(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
44-
#define LOG_WRN(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
45-
#define LOG_ERR(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
46-
#define LOG_DBG(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
43+
#if defined(LLAVA_LOG_OFF)
44+
# define LOG_INF(...)
45+
# define LOG_WRN(...)
46+
# define LOG_ERR(...)
47+
# define LOG_DBG(...)
48+
#else // defined(LLAVA_LOG_OFF)
49+
# define LOG_INF(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
50+
# define LOG_WRN(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
51+
# define LOG_ERR(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
52+
# define LOG_DBG(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
53+
#endif // defined(LLAVA_LOG_OFF)
4754

4855
//#define CLIP_DEBUG_FUNCTIONS
4956

examples/llava/llava.cpp

Lines changed: 19 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,17 @@
1111
#include <limits>
1212
#include <vector>
1313

14-
#define die(msg) do { fputs("error: " msg "\n", stderr); exit(1); } while (0)
15-
#define die_fmt(fmt, ...) do { fprintf(stderr, "error: " fmt "\n", __VA_ARGS__); exit(1); } while (0)
16-
17-
#define LOG_INF(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
18-
#define LOG_WRN(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
19-
#define LOG_ERR(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
20-
#define LOG_DBG(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
14+
#if defined(LLAVA_LOG_OFF)
15+
# define LOG_INF(...)
16+
# define LOG_WRN(...)
17+
# define LOG_ERR(...)
18+
# define LOG_DBG(...)
19+
#else // defined(LLAVA_LOG_OFF)
20+
# define LOG_INF(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
21+
# define LOG_WRN(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
22+
# define LOG_ERR(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
23+
# define LOG_DBG(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
24+
#endif // defined(LLAVA_LOG_OFF)
2125

2226
// RGB uint8 image
2327
struct clip_image_u8 {
@@ -498,10 +502,16 @@ static bool load_file_to_bytes(const char* path, unsigned char** bytesOut, long
498502
errno = 0;
499503
size_t ret = fread(buffer, 1, fileSize, file); // Read the file into the buffer
500504
if (ferror(file)) {
501-
die_fmt("read error: %s", strerror(errno));
505+
LOG_ERR("read error: %s", strerror(errno));
506+
free(buffer);
507+
fclose(file);
508+
return false;
502509
}
503510
if (ret != (size_t) fileSize) {
504-
die("unexpectedly reached end of file");
511+
LOG_ERR("unexpectedly reached end of file");
512+
free(buffer);
513+
fclose(file);
514+
return false;
505515
}
506516
fclose(file); // Close the file
507517

0 commit comments

Comments
 (0)