Skip to content

Commit b081001

Browse files
authored
Add capability to attach appropriate inner block size attribute (#701)
1 parent ebb7417 commit b081001

File tree

5 files changed

+355
-98
lines changed

5 files changed

+355
-98
lines changed

include/imex/Dialect/XeTile/Transforms/Passes.h

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
#ifndef _XeTile_PASSES_H_INCLUDED_
1717
#define _XeTile_PASSES_H_INCLUDED_
1818

19+
#include "imex/Utils/XeArch.h"
1920
#include <mlir/Pass/Pass.h>
2021

2122
namespace mlir {
@@ -35,15 +36,18 @@ class XeTypeConverter;
3536
//===----------------------------------------------------------------------===//
3637

3738
std::unique_ptr<mlir::Pass> createXeTileInitDuplicatePass();
38-
std::unique_ptr<mlir::Pass> createXeTileBlockingPass();
39+
40+
std::unique_ptr<mlir::Pass>
41+
createXeTileBlockingPass(const std::string &device = "pvc");
3942

4043
///
4144
void populateXeTileInitDuplicatePatterns(imex::XeTypeConverter &converter,
4245
mlir::RewritePatternSet &patterns);
4346

4447
///
4548
void populateXeTileBlockingPatterns(imex::XeTypeConverter &converter,
46-
mlir::RewritePatternSet &patterns);
49+
mlir::RewritePatternSet &patterns,
50+
std::shared_ptr<XeuArchInterface> ptruArch);
4751

4852
//===----------------------------------------------------------------------===//
4953
// Registration

include/imex/Dialect/XeTile/Transforms/Passes.td

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,10 @@ def XeTileBlocking : Pass<"xetile-blocking", "::mlir::gpu::GPUModuleOp">{
5454
"mlir::vector::VectorDialect"];
5555

5656
let options = [
57-
Option<"device", "device", "std::string", /*default=*/"\"default\"",
58-
"Specify the target device arch. Available examples are pvc. "
59-
"If not specified a default one will be used.">
60-
];
57+
Option<"device", "device", "std::string",
58+
/*default=*/"\"pvc\"",
59+
"gpu platform architecture where these ops are running">
60+
];
6161
}
6262

6363
#endif // _XeTile_PASSES_TD_INCLUDED_

0 commit comments

Comments
 (0)