diff options
Diffstat (limited to 'files/unit_test/unit_test.h')
-rw-r--r-- | files/unit_test/unit_test.h | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/files/unit_test/unit_test.h b/files/unit_test/unit_test.h index 0a8df4d2..99cc8d19 100644 --- a/files/unit_test/unit_test.h +++ b/files/unit_test/unit_test.h @@ -11,10 +11,10 @@ #ifndef UNIT_TEST_UNIT_TEST_H_ // NOLINT #define UNIT_TEST_UNIT_TEST_H_ +#include <stddef.h> // For NULL #ifdef _WIN32 #include <windows.h> #else -#include <sys/resource.h> #include <sys/time.h> #endif @@ -77,7 +77,18 @@ static inline bool SizeValid(int src_width, #define free_aligned_buffer_page_end(var) \ free(var##_mem); \ - var = 0 + var = NULL + +#define align_buffer_page_end_16(var, size) \ + uint8_t* var##_mem = \ + reinterpret_cast<uint8_t*>(malloc(((size)*2 + 4095 + 63) & ~4095)); \ + uint16_t* var = reinterpret_cast<uint16_t*>( \ + (intptr_t)(var##_mem + (((size)*2 + 4095 + 63) & ~4095) - (size)*2) & \ + ~63) + +#define free_aligned_buffer_page_end_16(var) \ + free(var##_mem); \ + var = NULL #ifdef WIN32 static inline double get_time() { |