GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-03 01:04:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix handmade_aligned_malloc offset computation.

Browse Source
This commit is contained in:
Antonio Sánchez 2022-10-27 17:33:47 +00:00 committed by Rasmus Munk Larsen
parent adb30efb25
commit ab407b2b6e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Eigen/src/Core/util/Memory.h
View File

@ -104,7 +104,7 @@ EIGEN_DEVICE_FUNC inline void* handmade_aligned_malloc(std::size_t size, std::si
eigen_assert(alignment >= sizeof(void*) && (alignment & (alignment-1)) == 0 && "Alignment must be at least sizeof(void*) and a power of 2"); eigen_assert(alignment >= sizeof(void*) && (alignment & (alignment-1)) == 0 && "Alignment must be at least sizeof(void*) and a power of 2");
void* original = std::malloc(size + alignment); void* original = std::malloc(size + alignment);
if (original == 0) return 0; if (original == 0) return 0;
uint8_t offset = alignment - reinterpret_cast<std::size_t>(original) & (alignment - 1); uint8_t offset = alignment - (reinterpret_cast<std::size_t>(original) & (alignment - 1));
void* aligned = static_cast<char*>(original) + offset; void* aligned = static_cast<char*>(original) + offset;
*(static_cast<char*>(aligned) - 1) = offset; *(static_cast<char*>(aligned) - 1) = offset;
return aligned; return aligned;
@ -133,7 +133,7 @@ EIGEN_DEVICE_FUNC inline void* handmade_aligned_realloc(void* ptr, std::size_t s
void* original = std::realloc(previous_original, size + alignment); void* original = std::realloc(previous_original, size + alignment);
if (original == 0) return 0; if (original == 0) return 0;
if (original != previous_original) { if (original != previous_original) {
uint8_t offset = alignment - reinterpret_cast<std::size_t>(original) & (alignment - 1); uint8_t offset = alignment - (reinterpret_cast<std::size_t>(original) & (alignment - 1));
void* aligned = static_cast<char*>(original) + offset; void* aligned = static_cast<char*>(original) + offset;
std::memmove(aligned, ptr, size); std::memmove(aligned, ptr, size);
*(static_cast<char*>(aligned) - 1) = offset; *(static_cast<char*>(aligned) - 1) = offset;