Code Review

Author: winskuo-quic

backends/qualcomm/aot/wrappers/TensorWrapper.cpp

@@ -7,7 +7,6 @@
  */
 
 #include <executorch/backends/qualcomm/aot/wrappers/TensorWrapper.h>
-#include <executorch/backends/qualcomm/runtime/backends/QnnCustomProtocol.h>
 
 #include <atomic>
 #include <cstring>

backends/qualcomm/aot/wrappers/TensorWrapper.h

@@ -19,6 +19,9 @@
 #define QNN_TENSOR_VER_PTR(x) (&((x).v2))
 #define QNN_OP_VER_PTR(x) (&((x).v1))
 
+// This is for direct mode, especially LPAI
+#define QNN_TENSOR_ALIGNMENT 64
+
 namespace executorch {
 namespace backends {
 namespace qnn {

backends/qualcomm/runtime/QnnExecuTorch.h

@@ -26,9 +26,6 @@
 #define QNN_RUNTIME_LPAI_AFFINITY "qnn_runtime_lpai_affinity"
 #define QNN_RUNTIME_LPAI_CORE_SELECTION "qnn_runtime_lpai_core_selection"
 
-// This is for direct mode, especially LPAI
-#define QNN_TENSOR_ALIGNMENT 64
-
 #ifdef __cplusplus
 extern "C" {
 #endif // __cplusplus

backends/qualcomm/runtime/backends/direct_mode/QnnExecuTorchIdlWrapper.cpp

@@ -6,6 +6,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 #include <executorch/backends/qualcomm/runtime/backends/direct_mode/QnnExecuTorchIdlWrapper.h>
+#include <executorch/backends/qualcomm/aot/wrappers/TensorWrapper.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <fstream>