From 6bc651ec4a32034a96760751a296541a66c4a8ab Mon Sep 17 00:00:00 2001
From: gongsu832 <gong_su@hotmail.com>
Date: Fri, 21 Jul 2023 16:24:08 -0400
Subject: [PATCH] Fix NNPA ZHigh Zlow dependency on OMLayoutHelper (#2384)

Signed-off-by: Gong Su <gong_su@hotmail.com>
---
 src/Accelerators/NNPA/Transform/ZHigh/CMakeLists.txt | 2 +-
 src/Accelerators/NNPA/Transform/ZLow/CMakeLists.txt  | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/Accelerators/NNPA/Transform/ZHigh/CMakeLists.txt b/src/Accelerators/NNPA/Transform/ZHigh/CMakeLists.txt
index 256d2921b8..95a9d79d87 100644
--- a/src/Accelerators/NNPA/Transform/ZHigh/CMakeLists.txt
+++ b/src/Accelerators/NNPA/Transform/ZHigh/CMakeLists.txt
@@ -7,8 +7,8 @@ add_onnx_mlir_library(OMZHighConstPropagation
   ZHighConstPropagation.cpp
 
   DEPENDS
+  OMLayoutHelper
   OMONNXZHighConstPropagationIncGen
-  libzdnn
 
   LINK_LIBS PUBLIC
   MLIRRewrite
diff --git a/src/Accelerators/NNPA/Transform/ZLow/CMakeLists.txt b/src/Accelerators/NNPA/Transform/ZLow/CMakeLists.txt
index 1bfe136f0b..d7b9417edc 100644
--- a/src/Accelerators/NNPA/Transform/ZLow/CMakeLists.txt
+++ b/src/Accelerators/NNPA/Transform/ZLow/CMakeLists.txt
@@ -3,6 +3,9 @@
 add_onnx_mlir_library(OMZLowRewrite
   ZLowRewrite.cpp
 
+  DEPENDS
+  OMLayoutHelper
+
   LINK_LIBS PUBLIC
   MLIRAffineUtils
   MLIRFuncDialect