1
0
Fork 0
mirror of https://github.com/sockspls/badfish synced 2025-05-02 01:29:36 +00:00

Use _mm_malloc on _WIN32

This commit is contained in:
Joost VandeVondele 2020-07-26 22:22:36 +02:00
parent a6c614da03
commit a8bdf69c71

View file

@ -41,10 +41,6 @@ typedef bool(*fun3_t)(HANDLE, CONST GROUP_AFFINITY*, PGROUP_AFFINITY);
} }
#endif #endif
#if defined(_MSC_VER)
#include <malloc.h>
#endif
#include <fstream> #include <fstream>
#include <iomanip> #include <iomanip>
#include <iostream> #include <iostream>
@ -305,8 +301,8 @@ void prefetch(void* addr) {
void* std_aligned_alloc(size_t alignment, size_t size) { void* std_aligned_alloc(size_t alignment, size_t size) {
#if defined(__APPLE__) #if defined(__APPLE__)
return aligned_alloc(alignment, size); return aligned_alloc(alignment, size);
#elif defined(_MSC_VER) #elif defined(_WIN32)
return _aligned_alloc(size, alignment); return _mm_malloc(size, alignment);
#else #else
return std::aligned_alloc(alignment, size); return std::aligned_alloc(alignment, size);
#endif #endif
@ -315,8 +311,8 @@ void* std_aligned_alloc(size_t alignment, size_t size) {
void std_aligned_free(void* ptr) { void std_aligned_free(void* ptr) {
#if defined(__APPLE__) #if defined(__APPLE__)
free(ptr); free(ptr);
#elif defined(_MSC_VER) #elif defined(_WIN32)
_aligned_free(ptr); _mm_free(ptr);
#else #else
free(ptr); free(ptr);
#endif #endif