[Mlir-commits] [mlir] [MLIR][ArmNeon] Add a couple of negative tests for BFMMLA with scalable dimensions (PR #145882)
Momchil Velikov
llvmlistbot at llvm.org
Thu Jun 26 05:51:27 PDT 2025
https://github.com/momchil-velikov created https://github.com/llvm/llvm-project/pull/145882
None
>From 9bbc8a2c6442884b9a8dcce9759f10bb2e4cb9b9 Mon Sep 17 00:00:00 2001
From: Momchil Velikov <momchil.velikov at arm.com>
Date: Thu, 26 Jun 2025 12:46:23 +0000
Subject: [PATCH] [MLIR][ArmNeon] Add a couple of negative tests for BFMMLA
with scalable dimensions
---
mlir/test/Dialect/ArmNeon/invalid.mlir | 20 ++++++++++++++++++++
1 file changed, 20 insertions(+)
diff --git a/mlir/test/Dialect/ArmNeon/invalid.mlir b/mlir/test/Dialect/ArmNeon/invalid.mlir
index 989293a3508b4..00d489b639191 100644
--- a/mlir/test/Dialect/ArmNeon/invalid.mlir
+++ b/mlir/test/Dialect/ArmNeon/invalid.mlir
@@ -114,6 +114,16 @@ func.func @bfmmla_invalid_dimension_lhs_rhs(%acc: vector<4xf32>,
// -----
+func.func @bfmmla_scalable_dimension_lhs_rhs(%acc: vector<4xf32>,
+ %lhs: vector<[8]xbf16>,
+ %rhs: vector<[8]xbf16>) -> vector<4xf32> {
+ // expected-error at +1 {{operand #1 must be a vector with length 8 of bfloat16 type values, but got 'vector<[8]xbf16>'}}
+ %0 = arm_neon.intr.bfmmla %acc, %lhs, %rhs : vector<[8]xbf16> to vector<4xf32>
+ return %0 : vector<4xf32>
+}
+
+// -----
+
func.func @bfmmla_invalid_element_type_acc(%acc: vector<4xi32>,
%lhs: vector<8xbf16>,
%rhs: vector<8xbf16>) -> vector<4xi32> {
@@ -131,3 +141,13 @@ func.func @bfmmla_invalid_dimension_acc(%acc: vector<8xf32>,
%0 = arm_neon.intr.bfmmla %acc, %lhs, %rhs : vector<8xbf16> to vector<8xf32>
return %0 : vector<8xf32>
}
+
+// -----
+
+func.func @bfmmla_scalable_dimension_acc(%acc: vector<[4]xf32>,
+ %lhs: vector<8xbf16>,
+ %rhs: vector<8xbf16>) -> vector<[4]xf32> {
+ // expected-error at +1 {{op operand #0 must be a vector with length 4 of 32-bit float values, but got 'vector<[4]xf32>'}}
+ %0 = arm_neon.intr.bfmmla %acc, %lhs, %rhs : vector<8xbf16> to vector<[4]xf32>
+ return %0 : vector<[4]xf32>
+}
More information about the Mlir-commits
mailing list