diff options
| author | Zhaoshi Zheng <zhaoshiz@quicinc.com> | 2024-07-23 21:52:22 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-23 21:52:22 -0700 |
| commit | 6942f1d5aa232face8269ce78c4de7d45571a8e9 (patch) | |
| tree | b1d31dfde414624d65906cb7aa4c5274549b9908 /lldb/source/Plugins/ScriptInterpreter/Python/Interfaces/ScriptedProcessPythonInterface | |
| parent | 2ca300f914b1bdeef5f08415bbcbf1803c186401 (diff) | |
[MLIR][Linalg] Scalable Vectorization of Reduction on the Trailing Dimension (#97788)
Allow scalable vectorization of linalg::reduce and linalg::generic that has
reduction iterator(s) with two restrictions:
1. The reduction dim is the last (innermost) dim of the op; and
2. Only the reduction dim is requested for scalable vectorization.
One exception is that scalable vectorization of the reduction dim in
Matmul-like ops are not supported even above restrictions are met.
Allowed combinations of scalable flags and iterator types:
Matmul:
Iterators: ["parallel", "parallel", "reduction"]
Scalable Flags: ["true", "true", "false"]
["false", "true", "false"]
Matvec:
Iterators: ["parallel", "reduction"]
Scalable Flags: ["false", "true"]
["true", "false"]
Diffstat (limited to 'lldb/source/Plugins/ScriptInterpreter/Python/Interfaces/ScriptedProcessPythonInterface')
0 files changed, 0 insertions, 0 deletions
