From d6b559c431b73a66a7ddacbd275cb4da7b430c1a Mon Sep 17 00:00:00 2001
From: Wenzhi Cui <wcui@google.com>
Date: Mon, 4 Mar 2024 00:45:04 +0000
Subject: [PATCH] Fix compilation failure due to missing header

---
 src/Burst.h                   | 1 +
 src/PIMCmd.h                  | 1 +
 src/tests/KernelTestCases.cpp | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Burst.h b/src/Burst.h
index 8009bf0..30ff3c2 100644
--- a/src/Burst.h
+++ b/src/Burst.h
@@ -20,6 +20,7 @@
 #include <sstream>
 #include <string>
 #include <vector>
+#include <iterator>
 
 #include "FP16.h"
 #include "npy.h"
diff --git a/src/PIMCmd.h b/src/PIMCmd.h
index 3af4839..5e7dc7c 100644
--- a/src/PIMCmd.h
+++ b/src/PIMCmd.h
@@ -13,6 +13,7 @@
 #ifndef __PIM_CMD_HPP__
 #define __PIM_CMD_HPP__
 
+#include <cstdint>
 #include <bitset>
 #include <iostream>
 #include <sstream>
diff --git a/src/tests/KernelTestCases.cpp b/src/tests/KernelTestCases.cpp
index 0ca6269..d230008 100644
--- a/src/tests/KernelTestCases.cpp
+++ b/src/tests/KernelTestCases.cpp
@@ -54,7 +54,7 @@ TEST_F(PIMKernelFixture, gemv_tree)
 
     delete result_;
     delete reduced_result_;
-    delete temp_fp16;
+    delete[] temp_fp16;
     delete dim_data;
 }