More progress on Triton=>TritonGPU conversion (works for matmul)

2022-05-09 21:19:53 +08:00
parent 0c5319eed9
commit 96876a46d1
3 changed files with 64 additions and 32 deletions
--- a/lib/Dialect/TritonGPU/IR/Dialect.cpp
+++ b/lib/Dialect/TritonGPU/IR/Dialect.cpp
@@ -41,7 +41,11 @@ TritonGPUSharedEncodingAttr::parse(mlir::AsmParser &parser, ::mlir::Type type) {
 }

 void TritonGPUSharedEncodingAttr::print(mlir::AsmPrinter &printer) const {
-  llvm_unreachable("Not implemented");
+  printer << "<"
+          // << "threadTileSize = " << getThreadTileSize()
+          // << ", blockTileSize = " << getBlockTileSize()
+          // << ", order = " << getOrder()
+          << ">";
 }

 void TritonGPUDialect::initialize() {