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 @@ -8,7 +8,7 @@ // Multiply into zeroed destination. func @kernel1(%arg0: memref<2x4xbf16>, %arg1: memref<2x4xbf16>, - %arg2: memref<2x2xf32>) { + %arg2: memref<2x2xf32>) { %0 = arith.constant 0 : index %1 = amx.tile_load %arg0[%0, %0] : memref<2x4xbf16> into vector<2x4xbf16> %2 = amx.tile_load %arg1[%0, %0] : memref<2x4xbf16> into vector<2x4xbf16> @@ -21,7 +21,7 @@ // Multiply and update into destination. func @kernel2(%arg0: memref<2x4xbf16>, %arg1: memref<2x4xbf16>, - %arg2: memref<2x2xf32>) { + %arg2: memref<2x2xf32>) { %0 = arith.constant 0 : index %1 = amx.tile_load %arg0[%0, %0] : memref<2x4xbf16> into vector<2x4xbf16> %2 = amx.tile_load %arg1[%0, %0] : memref<2x4xbf16> into vector<2x4xbf16> 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 @@ -8,7 +8,7 @@ // Multiply into zeroed destination. func @kernel1(%arg0: memref<2x8xi8>, %arg1: memref<2x8xi8>, - %arg2: memref<2x2xi32>) { + %arg2: memref<2x2xi32>) { %0 = arith.constant 0 : index %1 = amx.tile_load %arg0[%0, %0] : memref<2x8xi8> into vector<2x8xi8> %2 = amx.tile_load %arg1[%0, %0] : memref<2x8xi8> into vector<2x8xi8> @@ -21,7 +21,7 @@ // Multiply and update into destination. func @kernel2(%arg0: memref<2x8xi8>, %arg1: memref<2x8xi8>, - %arg2: memref<2x2xi32>) { + %arg2: memref<2x2xi32>) { %0 = arith.constant 0 : index %1 = amx.tile_load %arg0[%0, %0] : memref<2x8xi8> into vector<2x8xi8> %2 = amx.tile_load %arg1[%0, %0] : memref<2x8xi8> into vector<2x8xi8>