diff --git a/compiler/plugins/target/AMD-AIE/aievec/XLLVMDialect.h b/compiler/plugins/target/AMD-AIE/aievec/XLLVMDialect.h
index 34e0a5b2f..310fa36fd 100644
--- a/compiler/plugins/target/AMD-AIE/aievec/XLLVMDialect.h
+++ b/compiler/plugins/target/AMD-AIE/aievec/XLLVMDialect.h
@@ -14,6 +14,7 @@
 #ifndef AIE_DIALECT_XLLVM_XLLVMDIALECT_H
 #define AIE_DIALECT_XLLVM_XLLVMDIALECT_H
 
+#include "llvm/IR/IRBuilder.h"
 #include "llvm/IR/LLVMContext.h"
 #include "llvm/IR/Module.h"
 #include "llvm/IR/Type.h"
diff --git a/compiler/plugins/target/AMD-AIE/aievec/test/precanonicalization-aieml-llvmir.mlir b/compiler/plugins/target/AMD-AIE/aievec/test/precanonicalization-aieml-llvmir.mlir
index bae91b9c4..6685153ff 100644
--- a/compiler/plugins/target/AMD-AIE/aievec/test/precanonicalization-aieml-llvmir.mlir
+++ b/compiler/plugins/target/AMD-AIE/aievec/test/precanonicalization-aieml-llvmir.mlir
@@ -28,7 +28,7 @@ func.func @scalar_extsi_to_shape_cast_swap(%s: vector<16x2xi8>) -> vector<32xi32
 // CHECK-LABEL: @extsi_to_broadcast_swap(
 // CHECK-SAME: %[[VIN:.*]]: vector<8xi8>
 func.func @extsi_to_broadcast_swap(%v: vector<8xi8>) -> vector<4x8xi32> {
-    // CHECK: %[[ZV:.*]] = arith.constant dense<0> : vector<4x8xi8>
+    // CHECK: %[[ZV:.*]] = ub.poison : vector<4x8xi8>
     // CHECK: %[[I0:.*]] = vector.insert %[[VIN]], %[[ZV]] [0] : vector<8xi8> into vector<4x8xi8>
     // CHECK: %[[I1:.*]] = vector.insert %[[VIN]], %[[I0]] [1] : vector<8xi8> into vector<4x8xi8>
     // CHECK: %[[I2:.*]] = vector.insert %[[VIN]], %[[I1]] [2] : vector<8xi8> into vector<4x8xi8>
@@ -44,7 +44,7 @@ func.func @extsi_to_broadcast_swap(%v: vector<8xi8>) -> vector<4x8xi32> {
 // CHECK-LABEL: @broadcast_to_insert(
 // CHECK-SAME: %[[V:.*]]: vector<8xbf16>
 func.func @broadcast_to_insert(%v: vector<8xbf16>) -> vector<1x4x8xbf16> {
-    // CHECK: %[[ZV:.*]] = arith.constant dense<0.000000e+00> : vector<4x8xbf16>
+    // CHECK: %[[ZV:.*]] = ub.poison : vector<4x8xbf16>
     // CHECK: %[[I0:.*]] = vector.insert %[[V]], %[[ZV]] [0] : vector<8xbf16> into vector<4x8xbf16>
     // CHECK: %[[I1:.*]] = vector.insert %[[V]], %[[I0]] [1] : vector<8xbf16> into vector<4x8xbf16>
     // CHECK: %[[I2:.*]] = vector.insert %[[V]], %[[I1]] [2] : vector<8xbf16> into vector<4x8xbf16>
diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Test/transform_dialect/matmul_fill_spec_simple_pack.mlir b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Test/transform_dialect/matmul_fill_spec_simple_pack.mlir
index 449e1a0b9..d9b226839 100644
--- a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Test/transform_dialect/matmul_fill_spec_simple_pack.mlir
+++ b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Test/transform_dialect/matmul_fill_spec_simple_pack.mlir
@@ -171,7 +171,7 @@ module attributes { transform.with_named_sequence } {
 //       CHECK:        memref.dealloc %{{.*}} : memref<1x1x4x4x8x8xi32, 2>
 //       CHECK:     }
 //       CHECK:     iree_linalg_ext.unpack %{{.*}} outer_dims_perm = [0, 1, 3, 2] inner_dims_pos = [2, 3] inner_tiles = [4, 8] into %{{.*}} : (memref<1x1x4x8x4x8xi32, 2> memref<1x1x32x32xi32, strided<[4096, 4096, 64, 1], offset: ?>, 1>)
-//       CHECK:     memref.dealloc %alloc_7 : memref<1x1x4x8x4x8xi32, 2>
+//       CHECK:     memref.dealloc %{{.*}} : memref<1x1x4x8x4x8xi32, 2>
 //       CHECK:   }
 //       CHECK:   iree_linalg_ext.unpack %{{.*}} inner_dims_pos = [0, 1] inner_tiles = [64, 64] into %{{.*}} : (memref<1x1x64x64xi32, 1> memref<64x64xi32, strided<[2048, 1], offset: ?>, #hal.descriptor_type<storage_buffer>>)
 //       CHECK:   memref.dealloc %{{.*}} : memref<1x1x512x64xi32, 1>
diff --git a/third_party/iree b/third_party/iree
index e14d6cd36..756e9e661 160000
--- a/third_party/iree
+++ b/third_party/iree
@@ -1 +1 @@
-Subproject commit e14d6cd36c5cd7dfa51c07da3600a7226b4431bd
+Subproject commit 756e9e66138129df1bb28a2f2fac06058f976bcf