[Triton-MLIR] Minor fixes to enable fused-softmax and layer-norm tutorials (#835)

This commit is contained in:
Qingyi Liu
2022-11-09 10:18:56 +08:00
committed by GitHub
parent 2da71b2aaa
commit e517b58d59
4 changed files with 5 additions and 5 deletions

View File

@@ -125,7 +125,7 @@ std::string PTXBuilder::dump() const {
lines.push_back(exec->dump());
}
return strJoin(lines, "\r\n");
return strJoin(lines, "\n\t");
}
PTXInstrExecution &PTXInstrCommon::call(ArrayRef<Operand *> oprs) {

View File

@@ -1051,7 +1051,7 @@ struct LoadOpConversion
if (other) {
for (size_t ii = 0; ii < nWords; ++ii) {
PTXInstr &mov = *ptxBuilder.create<>("mov");
mov.o("u", width);
mov.o("u" + std::to_string(width));
size_t size = width / valueElemNbits;