retain header orders

nod-ai · Nov 1, 2024 · f050d94 · f050d94
1 parent 555ec3a
commit f050d94
Show file tree

Hide file tree

Showing 7 changed files with 12 additions and 14 deletions.
diff --git a/runtime/src/iree-amd-aie/aie_runtime/iree_aie_configure.cc b/runtime/src/iree-amd-aie/aie_runtime/iree_aie_configure.cc
@@ -20,11 +20,6 @@
 
 using Path = std::filesystem::path;
 
-extern "C" {
-#include "xaie_hwcfg.h"
-#include "xaiengine.h"
-}
-
 namespace mlir::iree_compiler::AMDAIE {
 
 FailureOr<XAie_DmaDesc> initDMADesc(const AMDAIEDeviceModel &deviceModel,

diff --git a/runtime/src/iree-amd-aie/aie_runtime/iree_aie_runtime.cc b/runtime/src/iree-amd-aie/aie_runtime/iree_aie_runtime.cc
@@ -14,7 +14,6 @@
 
 extern "C" {
 #include "xaiengine.h"
-#include "xaie_hwcfg.h"
 #undef s8
 #undef u8
 #undef u16
@@ -169,7 +168,6 @@ int AMDAIEDeviceModel::rows() const {
     return MLIRAIELegacy::rows(*this);
   assert(isNPUDevice(device) && "expected NPU device");
   return devInst.NumRows;
-
 }
 
 int AMDAIEDeviceModel::columns() const {

diff --git a/runtime/src/iree-amd-aie/aie_runtime/iree_aie_runtime.h b/runtime/src/iree-amd-aie/aie_runtime/iree_aie_runtime.h
@@ -8,16 +8,25 @@
 #define IREE_AIE_RUNTIME_H
 
 #include <optional>
+#include <ostream>
+#include <sstream>
 #include <tuple>
 #include <type_traits>
 
-#include "iree-amd-aie/aie_runtime/AMDAIEEnums.h"
 #include "llvm/ADT/Twine.h"
 #include "llvm/Support/Debug.h"
+#include "llvm/Support/FormattedStream.h"
 #include "macros.h"
 
+// clang-format off
+#include "iree-amd-aie/aie_runtime/AMDAIEEnums.h"
+// clang-format on
+
 extern "C" {
+
+#include "xaie_hwcfg.h"
 #include "xaiengine.h"
+#include "xaiengine/xaie_device_aieml.h"
 #undef s8
 #undef u8
 #undef u16

diff --git a/runtime/src/iree-amd-aie/aie_runtime/test/test_control_packets.cc b/runtime/src/iree-amd-aie/aie_runtime/test/test_control_packets.cc
@@ -47,7 +47,7 @@ clang-format on
 #include <cstdio>
 
 #include "interpreter_op_impl.h"
-#include "iree-amd-aie/aie_runtime/xaie_hwcfg.h"
+#include "iree-amd-aie/aie_runtime/iree_aie_runtime.h"
 
 #define DATA_SIZE 17
 #define WRRD_DATA_SIZE 0x02

diff --git a/runtime/src/iree-amd-aie/aie_runtime/test/test_packet_switch_mode.cc b/runtime/src/iree-amd-aie/aie_runtime/test/test_packet_switch_mode.cc
@@ -129,7 +129,7 @@ clang-format on
 #include <cstdio>
 
 #include "interpreter_op_impl.h"
-#include "iree-amd-aie/aie_runtime/xaie_hwcfg.h"
+#include "iree-amd-aie/aie_runtime/iree_aie_runtime.h"
 
 using namespace mlir::iree_compiler;
 using namespace mlir::iree_compiler::AMDAIE;

diff --git a/runtime/src/iree-amd-aie/aie_runtime/test/test_transaction.cc b/runtime/src/iree-amd-aie/aie_runtime/test/test_transaction.cc
@@ -46,9 +46,6 @@ clang-format on
 
 #include "interpreter_op_impl.h"
 #include "iree-amd-aie/aie_runtime/iree_aie_runtime.h"
-#include "iree-amd-aie/aie_runtime/xaie_hwcfg.h"
-
-
 
 using namespace mlir::iree_compiler;
 using namespace mlir::iree_compiler::AMDAIE;

diff --git a/runtime/src/iree-amd-aie/aie_runtime/test/utest.cc b/runtime/src/iree-amd-aie/aie_runtime/test/utest.cc
@@ -230,7 +230,6 @@ clang-format off
 #include <string>
 
 #include "iree-amd-aie/aie_runtime/iree_aie_runtime.h"
-#include "iree-amd-aie/aie_runtime/xaie_hwcfg.h"
 
 int main(int argc, char** argv) {
   std::string elfPath(argv[1]);