[FIX] cast and include

tud-zih-energy · Nov 25, 2024 · 13202bd · 13202bd
1 parent b4618e5
commit 13202bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/include/firestarter/Environment/X86/Payload/AVX512Payload.hpp b/include/firestarter/Environment/X86/Payload/AVX512Payload.hpp
@@ -20,7 +20,7 @@
  *****************************************************************************/
 
 #pragma once
-
+#include <immintrin.h>
 #include "firestarter/Environment/X86/Payload/X86Payload.hpp"
 namespace firestarter::environment::x86::payload {
 

diff --git a/src/firestarter/Environment/X86/Payload/AVX512Payload.cpp b/src/firestarter/Environment/X86/Payload/AVX512Payload.cpp
@@ -189,14 +189,14 @@ auto AVX512Payload::compilePayload(const environment::payload::PayloadSettings&
 
   __bfloat16* src1;
   __bfloat16* src2;
-  uintptr_t src3;
+  void* src3;
   unsigned int aligned_alloc_size = static_cast<unsigned int>(static_cast<uint32_t>(MaxSize::ELEMENTS) * sizeof(__bfloat16));
   if (aligned_alloc_size % 1024) { // aligned_alloc expects size to be multiple of alignment (aka 1024)
     aligned_alloc_size = aligned_alloc_size + (1024 - (aligned_alloc_size % 1024));
   }
   src1 = static_cast<__bfloat16*>(aligned_alloc(1024, aligned_alloc_size));
   src2 = static_cast<__bfloat16*>(aligned_alloc(1024, aligned_alloc_size));
-  src3 = static_cast<uint64_t>(aligned_alloc(1024, aligned_alloc_size));
+  src3 = static_cast<void*>(aligned_alloc(1024, aligned_alloc_size));
   if ((static_cast<void*>(src1) == nullptr) || static_cast<void*>(src2) == nullptr ||
       static_cast<void*>(src3) == nullptr) { // uintptr_t garantuees we can cast it to void* and back
     std::cout << "[ERROR]: Allocation of source and target buffer for AMX failed. Aborting...\n";