gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more TritonGPU unit tests

Browse Source
This commit is contained in:
Yan Da
2022-06-05 14:25:09 +08:00
parent 55cf9a0a97
commit d5eca56cf3
3 changed files with 34 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/Dialect/TritonGPU/IR/Dialect.cpp
View File

@@ -102,12 +102,12 @@ TritonGPUShardedEncodingAttr::parse(AsmParser &parser, Type type) {
}
void TritonGPUShardedEncodingAttr::print(mlir::AsmPrinter &printer) const {
printer << "<"
printer << "<{"
<< "threadTileSize = [" << getThreadTileSize() << "]"
<< ", warpTileSize = [" << getWarpTileSize() << "]"
<< ", blockTileSize = [" << getBlockTileSize() << "]"
<< ", order = [" << getOrder() << "]"
<< ">";
<< "}>";
}
Attribute
@@ -175,11 +175,11 @@ TritonGPUSharedEncodingAttr::parse(AsmParser &parser, Type type) {
}
void TritonGPUSharedEncodingAttr::print(AsmPrinter &printer) const {
printer << "<"
printer << "<{"
<< "vec = " << getVec()
<< ", perPhase = " << getPerPhase()
<< ", order = [" << getOrder() << "]"
<< ">";
<< "}>";
}
void TritonGPUDialect::initialize() {