Skip to content

Commit 4bc6740

Browse files
hanhanWkeshavvinayak01
authored andcommitted
[DT][NFCI] Switch SetEncoding pass to walk-based pass. (iree-org#21662)
- Move the requirements to `isSupportedContractionOp`. - Convert `SetContractionOpEncoding::matchAndRewrite` to `setDataTilingEncodings` method. - Collect all the candidates, iterate and set encoding for each candidate. - Refactor `postPatterns` for all the strategies. It is a step towards iree-org#21246 --------- Signed-off-by: hanhanW <[email protected]> Signed-off-by: keshavvinayak01 <[email protected]>
1 parent a632a2e commit 4bc6740

File tree

2 files changed

+159
-169
lines changed

2 files changed

+159
-169
lines changed

compiler/src/iree/compiler/DispatchCreation/Passes.td

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,10 @@ def PropagateEncodingsPass :
329329
def SetEncodingPass : InterfacePass<"iree-dispatch-creation-set-encoding",
330330
"mlir::FunctionOpInterface"> {
331331
let summary = "Introduces tensor encoding for flow dispatch regions.";
332+
let description = [{
333+
Sets the encoding for compute operations to allow execution of the
334+
operations in tiled/padded layouts.
335+
}];
332336
let dependentDialects = [
333337
"mlir::linalg::LinalgDialect",
334338
"IREE::Flow::FlowDialect",

0 commit comments

Comments
 (0)