Skip to content

Commit c7feb0d

Browse files
author
Michael Skvortsov
committed
Minor improvement
1 parent 3876755 commit c7feb0d

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

llvm/lib/Target/TVM/TVMAsmPrinter.cpp

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -163,14 +163,15 @@ void TVMAsmPrinter::EmitFunctionBodyEnd() {
163163
auto *FI = MF->getInfo<TVMFunctionInfo>();
164164
unsigned Arguments = FI->getParams().size();
165165
unsigned ReturnValues = FI->getResults().size();
166+
const unsigned SmallEncodingLimit = 16;
166167

167168
if (Arguments > 0) {
168-
if (Blocks <= 16 && Arguments <= 16) {
169-
OutStreamer->EmitRawText(" BLKSWAP\t" + Twine(Arguments) + ", " +
169+
if (Blocks <= SmallEncodingLimit && Arguments <= SmallEncodingLimit) {
170+
OutStreamer->EmitRawText(" BLKSWAP " + Twine(Arguments) + ", " +
170171
Twine(Blocks));
171172
} else {
172-
OutStreamer->EmitRawText(" PUSHINT\t" + Twine(Arguments));
173-
OutStreamer->EmitRawText(" PUSHINT\t" + Twine(Blocks));
173+
OutStreamer->EmitRawText(" PUSHINT " + Twine(Arguments));
174+
OutStreamer->EmitRawText(" PUSHINT " + Twine(Blocks));
174175
OutStreamer->EmitRawText(" BLKSWX");
175176
}
176177
}
@@ -179,20 +180,20 @@ void TVMAsmPrinter::EmitFunctionBodyEnd() {
179180
OutStreamer->EmitRawText(" EXECUTE");
180181

181182
if (ReturnValues > 0) {
182-
if (Blocks <= 16 && ReturnValues <= 16) {
183-
OutStreamer->EmitRawText(" BLKSWAP\t" + Twine(Blocks) + ", " +
183+
if (Blocks <= SmallEncodingLimit && ReturnValues <= SmallEncodingLimit) {
184+
OutStreamer->EmitRawText(" BLKSWAP " + Twine(Blocks) + ", " +
184185
Twine(ReturnValues));
185186
} else {
186-
OutStreamer->EmitRawText(" PUSHINT\t" + Twine(Blocks));
187-
OutStreamer->EmitRawText(" PUSHINT\t" + Twine(ReturnValues));
187+
OutStreamer->EmitRawText(" PUSHINT " + Twine(Blocks));
188+
OutStreamer->EmitRawText(" PUSHINT " + Twine(ReturnValues));
188189
OutStreamer->EmitRawText(" BLKSWX");
189190
}
190191
}
191192

192-
if (Blocks < 16) {
193-
OutStreamer->EmitRawText(" BLKDROP\t" + Twine(Blocks));
193+
if (Blocks < SmallEncodingLimit) {
194+
OutStreamer->EmitRawText(" BLKDROP " + Twine(Blocks));
194195
} else {
195-
OutStreamer->EmitRawText(" PUSHINT\t" + Twine(Blocks));
196+
OutStreamer->EmitRawText(" PUSHINT " + Twine(Blocks));
196197
OutStreamer->EmitRawText(" DROPX");
197198
}
198199
}

0 commit comments

Comments
 (0)