[Mlir-commits] [mlir] [mlir][ArmSME] Add option to only enable streaming mode for scalable code (PR #94759)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Fri Jun 7 07:52:01 PDT 2024
================
@@ -116,10 +116,14 @@ def EnableArmStreaming
"not be used for input and/or output and the "
"function must return with ZA unchanged")
)}]>,
- Option<"onlyIfRequiredByOps", "only-if-required-by-ops", "bool",
+ Option<"ifRequiredByOps", "if-required-by-ops", "bool",
/*default=*/"false",
- "Only apply the selected streaming/ZA modes if the function "
- " contains ops that require them.">
+ "Apply the selected streaming/ZA modes if the function contains ops "
+ "that require them.">,
+ Option<"ifContainsScalableVectors", "if-contains-scalable-vectors",
+ "bool", /*default=*/"false",
+ "Apply the selected streaming/ZA modes if the function contains "
+ "operations that use scalable vector types.">
----------------
banach-space wrote:
The distinction between ought to be clearer. Here's a suggestion:
**ifRequredByOps**
> Apply the selected streaming/ZA modes if the function contains ops that require them (e.g. don't enable ZA, even when requested, when there are no ops operating on 2d scalable vectors).
**ifContainsScalableVectors"**
> Apply the selected streaming/ZA modes if the function contains operations that use scalable vector types (e.g. don't enable SSVE and/or ZA, even when requested, when there are no scalable vectors).
WDYT? (I've only really added an example)
https://github.com/llvm/llvm-project/pull/94759
More information about the Mlir-commits
mailing list