From 7ba244b3e8baa81931a7780f7a09b8b472dc1b44 Mon Sep 17 00:00:00 2001
From: Kunshang Ji <kunshang.ji@intel.com>
Date: Mon, 4 Sep 2023 00:43:23 +0000
Subject: [PATCH] add memalign 0 byte check

---
 ggml.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ggml.c b/ggml.c
index 04f938db3..a5bd7f4db 100644
--- a/ggml.c
+++ b/ggml.c
@@ -195,12 +195,12 @@ typedef void * thread_ret_t;
 #define GGML_ALIGNED_FREE(ptr)    _aligned_free(ptr)
 #else
 inline static void * ggml_aligned_malloc(size_t size) {
-    void * aligned_memory = NULL;
-#ifdef GGML_USE_CPU_HBM
     if (size == 0) {
-        GGML_PRINT("WARNING: Behavior may be unexpected when allocate 0 byte for hbw_posix_memalign!");
+        GGML_PRINT("WARNING: Behavior may be unexpected when allocate 0 byte for ggml_aligned_malloc!");
         return NULL;
     }
+    void * aligned_memory = NULL;
+#ifdef GGML_USE_CPU_HBM
     int result = hbw_posix_memalign(&aligned_memory, 16, size);
 #elif GGML_USE_METAL
     int result = posix_memalign(&aligned_memory, sysconf(_SC_PAGESIZE), size);