[Mlir-commits] [mlir] [mlir][ArmSME] Support widening outer products (PR #78975)
Cullen Rhodes
llvmlistbot at llvm.org
Wed Jan 24 02:43:10 PST 2024
================
@@ -814,6 +814,649 @@ let arguments = (ins
}];
}
+class OuterProductWideBase<string mnemonic,
+ list<Type> allowedInputVectorTypes,
+ list<Type> allowedResultVectorTypes,
+ int numOuterProducts> :
+ ArmSME_Op<mnemonic, [
+ ArmSMETileOpInterface,
+ AttrSizedOperandSegments,
+ AllTypesMatch<["lhs", "rhs"]>,
----------------
c-rhodes wrote:
> How about checking that element types mismatch? Perhaps I've missed it 🤔
do you mean to check that the element types of the inputs don't match the element type of the output?
https://github.com/llvm/llvm-project/pull/78975
More information about the Mlir-commits
mailing list