diff --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf.mlir
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf.mlir
@@ -32,7 +32,7 @@
   return
 }
 
-func @entry() {
+func @entry() -> i32 {
   %f0 = constant 0.0: f32
   %c0 = constant 0: index
   %c1 = constant 1: index
@@ -80,5 +80,6 @@
   memref.dealloc %b : memref<2x4xbf16>
   memref.dealloc %c : memref<2x2xf32>
 
-  return
+  %i0 = constant 0 : i32
+  return %i0 : i32
 }
diff --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-ext.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-ext.mlir
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-ext.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-ext.mlir
@@ -66,7 +66,7 @@
   return
 }
 
-func @entry() {
+func @entry() -> i32 {
   %c0 = constant 0: index
 
   // Set up memory.
@@ -194,5 +194,6 @@
   memref.dealloc %b : memref<4x16xi8>
   memref.dealloc %c : memref<16x4xi32>
 
-  return
+  %i0 = constant 0 : i32
+  return %i0 : i32
 }
diff --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli.mlir
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli.mlir
@@ -32,7 +32,7 @@
   return
 }
 
-func @entry() {
+func @entry() -> i32 {
   %i0 = constant 0: i32
   %c0 = constant 0: index
   %c1 = constant 1: index
@@ -80,5 +80,5 @@
   memref.dealloc %b : memref<2x8xi8>
   memref.dealloc %c : memref<2x2xi32>
 
-  return
+  return %i0 : i32
 }
diff --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero-block.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero-block.mlir
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero-block.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero-block.mlir
@@ -34,7 +34,7 @@
   return
 }
 
-func @entry() {
+func @entry() -> i32 {
   %f1  = constant 1.0: f32
   %c0  = constant 0: index
   %c1  = constant 1: index
@@ -78,5 +78,6 @@
   // Release resources.
   memref.dealloc %a : memref<4x32xf32>
 
-  return
+  %i0 = constant 0 : i32
+  return %i0 : i32
 }
diff --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero.mlir
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-tilezero.mlir
@@ -12,7 +12,7 @@
   return
 }
 
-func @entry() {
+func @entry() -> i32 {
   %i0 = constant 0: i32
   %i1 = constant 1: i32
   %c0 = constant 0: index
@@ -93,5 +93,5 @@
   // Release resources.
   memref.dealloc %a : memref<?x?xi32>
 
-  return
+  return %i0 : i32
 }