From: Dmitry Preobrazhensky Date: Mon, 27 May 2019 14:08:43 +0000 (+0000) Subject: [AMDGPU][MC] Enabled constant expressions as operands of s_waitcnt X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=a125c3ae9ad7f783784a607dc884f97bbd7957a5;p=llvm [AMDGPU][MC] Enabled constant expressions as operands of s_waitcnt See bug 40820: https://bugs.llvm.org/show_bug.cgi?id=40820 Reviewers: artem.tamazov, arsenm Differential Revision: https://reviews.llvm.org/D61017 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@361763 91177308-0d34-0410-b5e6-96231b3b80d8 --- diff --git a/lib/Target/AMDGPU/AsmParser/AMDGPUAsmParser.cpp b/lib/Target/AMDGPU/AsmParser/AMDGPUAsmParser.cpp index b55facadf49..bc7068ef756 100644 --- a/lib/Target/AMDGPU/AsmParser/AMDGPUAsmParser.cpp +++ b/lib/Target/AMDGPU/AsmParser/AMDGPUAsmParser.cpp @@ -4391,20 +4391,18 @@ encodeCnt( } bool AMDGPUAsmParser::parseCnt(int64_t &IntVal) { - StringRef CntName = Parser.getTok().getString(); - int64_t CntVal; - Parser.Lex(); - if (getLexer().isNot(AsmToken::LParen)) - return true; + SMLoc CntLoc = getLoc(); + StringRef CntName = getTokenStr(); - Parser.Lex(); - if (getLexer().isNot(AsmToken::Integer)) - return true; + if (!skipToken(AsmToken::Identifier, "expected a counter name") || + !skipToken(AsmToken::LParen, "expected a left parenthesis")) + return false; - SMLoc ValLoc = Parser.getTok().getLoc(); - if (getParser().parseAbsoluteExpression(CntVal)) - return true; + int64_t CntVal; + SMLoc ValLoc = getLoc(); + if (!parseExpr(CntVal)) + return false; AMDGPU::IsaVersion ISA = AMDGPU::getIsaVersion(getSTI().getCPU()); @@ -4417,49 +4415,43 @@ bool AMDGPUAsmParser::parseCnt(int64_t &IntVal) { Failed = encodeCnt(ISA, IntVal, CntVal, Sat, encodeExpcnt, decodeExpcnt); } else if (CntName == "lgkmcnt" || CntName == "lgkmcnt_sat") { Failed = encodeCnt(ISA, IntVal, CntVal, Sat, encodeLgkmcnt, decodeLgkmcnt); + } else { + Error(CntLoc, "invalid counter name " + CntName); + return false; } if (Failed) { Error(ValLoc, "too large value for " + CntName); - return true; + return false; } - if (getLexer().isNot(AsmToken::RParen)) { - return true; - } + if (!skipToken(AsmToken::RParen, "expected a closing parenthesis")) + return false; - Parser.Lex(); - if (getLexer().is(AsmToken::Amp) || getLexer().is(AsmToken::Comma)) { - const AsmToken NextToken = getLexer().peekTok(); - if (NextToken.is(AsmToken::Identifier)) { - Parser.Lex(); + if (trySkipToken(AsmToken::Amp) || trySkipToken(AsmToken::Comma)) { + if (isToken(AsmToken::EndOfStatement)) { + Error(getLoc(), "expected a counter name"); + return false; } } - return false; + return true; } OperandMatchResultTy AMDGPUAsmParser::parseSWaitCntOps(OperandVector &Operands) { AMDGPU::IsaVersion ISA = AMDGPU::getIsaVersion(getSTI().getCPU()); int64_t Waitcnt = getWaitcntBitMask(ISA); - SMLoc S = Parser.getTok().getLoc(); - - switch(getLexer().getKind()) { - default: return MatchOperand_ParseFail; - case AsmToken::Integer: - // The operand can be an integer value. - if (getParser().parseAbsoluteExpression(Waitcnt)) - return MatchOperand_ParseFail; - break; + SMLoc S = getLoc(); - case AsmToken::Identifier: - do { - if (parseCnt(Waitcnt)) - return MatchOperand_ParseFail; - } while(getLexer().isNot(AsmToken::EndOfStatement)); - break; + // If parse failed, do not return error code + // to avoid excessive error messages. + if (isToken(AsmToken::Identifier) && peekToken().is(AsmToken::LParen)) { + while (parseCnt(Waitcnt) && !isToken(AsmToken::EndOfStatement)); + } else { + parseExpr(Waitcnt); } + Operands.push_back(AMDGPUOperand::CreateImm(this, Waitcnt, S)); return MatchOperand_Success; } diff --git a/test/MC/AMDGPU/sopp-err.s b/test/MC/AMDGPU/sopp-err.s index be655c81805..c6df9df0090 100644 --- a/test/MC/AMDGPU/sopp-err.s +++ b/test/MC/AMDGPU/sopp-err.s @@ -100,7 +100,31 @@ s_waitcnt vmcnt(0xFFFFFFFFFFFF0000) // GCN: error: too large value for vmcnt s_waitcnt vmcnt(0), expcnt(0), lgkmcnt(0), -// GCN: error: failed parsing operand +// GCN: error: expected a counter name s_waitcnt vmcnt(0) & expcnt(0) & lgkmcnt(0)& -// GCN: error: failed parsing operand +// GCN: error: expected a counter name + +s_waitcnt vmcnt(0) & expcnt(0) & x +// GCN: error: expected a left parenthesis + +s_waitcnt vmcnt(0) & expcnt(0) x +// GCN: error: expected a left parenthesis + +s_waitcnt vmcnt(0) & expcnt(0) & 1 +// GCN: error: expected a counter name + +s_waitcnt vmcnt(0) & expcnt(0) 1 +// GCN: error: expected a counter name + +s_waitcnt vmcnt(0) & expcnt(0) x(0) +// GCN: error: invalid counter name x + +s_waitcnt vmcnt(x) +// GCN: error: expected absolute expression + +s_waitcnt x +// GCN: error: expected absolute expression + +s_waitcnt vmcnt(0 +// GCN: error: expected a closing parenthesis diff --git a/test/MC/AMDGPU/sopp.s b/test/MC/AMDGPU/sopp.s index 3c426c08a0b..807a6d469a5 100644 --- a/test/MC/AMDGPU/sopp.s +++ b/test/MC/AMDGPU/sopp.s @@ -131,6 +131,30 @@ s_waitcnt lgkmcnt_sat(15) s_waitcnt lgkmcnt_sat(16) // GCN: s_waitcnt ; encoding: [0x7f,0x0f,0x8c,0xbf] +x=1 +s_waitcnt lgkmcnt_sat(x+1) +// GCN: s_waitcnt lgkmcnt(2) ; encoding: [0x7f,0x02,0x8c,0xbf] + +s_waitcnt lgkmcnt_sat(1+x) +// GCN: s_waitcnt lgkmcnt(2) ; encoding: [0x7f,0x02,0x8c,0xbf] + +s_waitcnt x+1 +// GCN: s_waitcnt vmcnt(2) expcnt(0) lgkmcnt(0) ; encoding: [0x02,0x00,0x8c,0xbf] + +s_waitcnt 1+x +// GCN: s_waitcnt vmcnt(2) expcnt(0) lgkmcnt(0) ; encoding: [0x02,0x00,0x8c,0xbf] + +lgkmcnt_sat=1 +s_waitcnt lgkmcnt_sat +// GCN: s_waitcnt vmcnt(1) expcnt(0) lgkmcnt(0) ; encoding: [0x01,0x00,0x8c,0xbf] + +s_waitcnt lgkmcnt_sat+1 +// GCN: s_waitcnt vmcnt(2) expcnt(0) lgkmcnt(0) ; encoding: [0x02,0x00,0x8c,0xbf] + +//===----------------------------------------------------------------------===// +// misc sopp instructions +//===----------------------------------------------------------------------===// + s_sethalt 9 // GCN: s_sethalt 9 ; encoding: [0x09,0x00,0x8d,0xbf]