fix: hbw_posix_memalign alignment

giladgd · giladgd · commit ecddb2452ab0 · 2024-10-16T02:03:03.000+03:00
diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
@@ -395,7 +395,7 @@ void * ggml_aligned_malloc(size_t size) {
     }
     void * aligned_memory = NULL;
 #ifdef GGML_USE_CPU_HBM
-    int result = hbw_posix_memalign(&aligned_memory, 16, size);
+    int result = hbw_posix_memalign(&aligned_memory, TENSOR_ALIGNMENT, size);
 #elif TARGET_OS_OSX
     kern_return_t alloc_status = vm_allocate((vm_map_t) mach_task_self(), (vm_address_t *) &aligned_memory, size, VM_FLAGS_ANYWHERE);
     int result = EFAULT;

Original file line number	Diff line number	Diff line change
`@@ -395,7 +395,7 @@ void * ggml_aligned_malloc(size_t size) {`
`395`	`395`	`}`
`396`	`396`	`void * aligned_memory = NULL;`
`397`	`397`	`#ifdef GGML_USE_CPU_HBM`
`398`		`- int result = hbw_posix_memalign(&aligned_memory, 16, size);`
	`398`	`+ int result = hbw_posix_memalign(&aligned_memory, TENSOR_ALIGNMENT, size);`
`399`	`399`	`#elif TARGET_OS_OSX`
`400`	`400`	`kern_return_t alloc_status = vm_allocate((vm_map_t) mach_task_self(), (vm_address_t *) &aligned_memory, size, VM_FLAGS_ANYWHERE);`
`401`	`401`	`int result = EFAULT;`