xilpm: versal: server: Fix bug in AIE2 zeroization

There is a bug in AIE2 zeriozation function when polling for memory
zeroization complete. Currently the entire memory register is being
checked against zero but instead we need to check the bits specific
to the memory tiles. This patch updates the zeroization check by
adding a mask so that only the desired bits are checked for zero.

Signed-off-by: Nicole Baze <[email protected]>
Acked-by: Jesus De Haro <[email protected]>

Author: Nicole Baze

lib/sw_services/xilpm/src/versal/server/xpm_aie.c

@@ -2030,16 +2030,16 @@ static XStatus Aie2_Zeroization(const XPm_Device *AieDev, u32 ColStart, u32 ColE
 	       (XST_SUCCESS != CoreZeroStatus) ||
 	       (XST_SUCCESS != MemZeroStatus)) {
 
-		if (0U == AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Mrow) +
-				    AIE2_MEM_TILE_MODULE_MEM_CTRL_OFFSET)) {
+		if (0U == (AIE2_MEM_TILE_MODULE_MEM_CTRL_MEM_ZEROISATION_MASK &
+			   (AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Mrow) + AIE2_MEM_TILE_MODULE_MEM_CTRL_OFFSET)))) {
 			MemTileZeroStatus = XST_SUCCESS;
 		}
-		if (0U == AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Row) +
-				    AIE2_CORE_MODULE_MEM_CTRL_OFFSET)) {
+		if (0U == (AIE2_CORE_MODULE_MEM_CTRL_MEM_ZEROISATION_MASK &
+			   (AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Row) + AIE2_CORE_MODULE_MEM_CTRL_OFFSET)))) {
 			CoreZeroStatus = XST_SUCCESS;
 		}
-		if (0U == AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Row) +
-				    AIE2_MEM_MODULE_MEM_CTRL_OFFSET)) {
+		if (0U == (AIE2_MEM_MODULE_MEM_CTRL_MEM_ZEROISATION_MASK &
+			   (AieRead64(AIE2_TILE_BADDR(NocAddress, Col, Row) + AIE2_MEM_MODULE_MEM_CTRL_OFFSET)))) {
 			MemZeroStatus = XST_SUCCESS;
 		}