LiuChen3 updated this revision to Diff 304730.
LiuChen3 added a comment.
Rebase.
Adding the '{}' to prefix when generate IR.
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D90441/new/
https://reviews.llvm.org/D90441
Files:
clang/lib/AST/Stmt.cpp
clang/test/CodeGen/X86/ms-inline-asm-prefix.c
llvm/lib/Target/X86/AsmParser/X86AsmParser.cpp
Index: llvm/lib/Target/X86/AsmParser/X86AsmParser.cpp
===================================================================
--- llvm/lib/Target/X86/AsmParser/X86AsmParser.cpp
+++ llvm/lib/Target/X86/AsmParser/X86AsmParser.cpp
@@ -3064,7 +3064,32 @@
}
continue;
}
-
+ // Parse MASM style pseudo prefixes.
+ if (isParsingMSInlineAsm()) {
+ bool IsPrefix = false;
+ if (Name == "vex") {
+ ForcedVEXEncoding = VEXEncoding_VEX;
+ IsPrefix = true;
+ } else if (Name == "vex2") {
+ ForcedVEXEncoding = VEXEncoding_VEX2;
+ IsPrefix = true;
+ } else if (Name == "vex3") {
+ ForcedVEXEncoding = VEXEncoding_VEX3;
+ IsPrefix = true;
+ } else if (Name == "evex") {
+ ForcedVEXEncoding = VEXEncoding_EVEX;
+ IsPrefix = true;
+ }
+ if (IsPrefix) {
+ NameLoc = Parser.getTok().getLoc();
+ if (getLexer().isNot(AsmToken::Identifier))
+ return Error(Parser.getTok().getLoc(), "Expected identifier");
+ // FIXME: The mnemonic won't match correctly if its not in lower case.
+ Name = Parser.getTok().getString();
+ Parser.Lex();
+ continue;
+ }
+ }
break;
}
@@ -4370,10 +4395,16 @@
MCInst Inst;
- // If VEX3 encoding is forced, we need to pass the USE_VEX3 flag to the
- // encoder.
- if (ForcedVEXEncoding == VEXEncoding_VEX3)
+ // If VEX/EVEX encoding is forced, we need to pass the USE_* flag to the
+ // encoder and printer.
+ if (ForcedVEXEncoding == VEXEncoding_VEX)
+ Prefixes |= X86::IP_USE_VEX;
+ else if (ForcedVEXEncoding == VEXEncoding_VEX2)
+ Prefixes |= X86::IP_USE_VEX2;
+ else if (ForcedVEXEncoding == VEXEncoding_VEX3)
Prefixes |= X86::IP_USE_VEX3;
+ else if (ForcedVEXEncoding == VEXEncoding_EVEX)
+ Prefixes |= X86::IP_USE_EVEX;
// Set encoded flags for {disp8} and {disp32}.
if (ForcedDispEncoding == DispEncoding_Disp8)
Index: clang/test/CodeGen/X86/ms-inline-asm-prefix.c
===================================================================
--- /dev/null
+++ clang/test/CodeGen/X86/ms-inline-asm-prefix.c
@@ -0,0 +1,20 @@
+// REQUIRES: x86-registered-target
+// RUN:%clang_cc1 %s -ferror-limit 0 -triple=x86_64-pc-windows-msvc -target-feature +avx512f -target-feature +avx2 -target-feature +avx512vl -fasm-blocks -mllvm -x86-asm-syntax=intel -S -o - | FileCheck %s -check-prefix=INTEL
+// RUN:%clang_cc1 %s -ferror-limit 0 -triple=x86_64-pc-windows-msvc -target-feature +avx512f -target-feature +avx2 -target-feature +avx512vl -fasm-blocks -mllvm -x86-asm-syntax=att -S -o - | FileCheck %s -check-prefix=ATT
+
+void check_inline_prefix(void) {
+ __asm {
+ // INTEL: {vex} vcvtps2pd xmm0, xmm1
+ // INTEL: {vex2} vcvtps2pd xmm0, xmm1
+ // INTEL: {vex3} vcvtps2pd xmm0, xmm1
+ // INTEL: {evex} vcvtps2pd xmm0, xmm1
+ // ATT: {vex} vcvtps2pd %xmm1, %xmm0
+ // ATT: {vex2} vcvtps2pd %xmm1, %xmm0
+ // ATT: {vex3} vcvtps2pd %xmm1, %xmm0
+ // ATT: {evex} vcvtps2pd %xmm1, %xmm0
+ vex vcvtps2pd xmm0, xmm1
+ vex2 vcvtps2pd xmm0, xmm1
+ vex3 vcvtps2pd xmm0, xmm1
+ evex vcvtps2pd xmm0, xmm1
+ }
+}
Index: clang/lib/AST/Stmt.cpp
===================================================================
--- clang/lib/AST/Stmt.cpp
+++ clang/lib/AST/Stmt.cpp
@@ -791,7 +791,27 @@
/// Assemble final IR asm string (MS-style).
std::string MSAsmStmt::generateAsmString(const ASTContext &C) const {
// FIXME: This needs to be translated into the IR string representation.
- return std::string(AsmStr);
+ SmallVector<StringRef, 8> Pieces;
+ AsmStr.split(Pieces, "\n\t");
+ std::string MSAsmString;
+ for (size_t i = 0, e = Pieces.size(); i < e; ++i) {
+ StringRef Instruction = Pieces[i];
+ // For vex/vex2/vex3/evex masm style prefix, convert it to att style
+ // since we don't support masm style prefix in backend.
+ if (Instruction.startswith("vex "))
+ MSAsmString += '{' + Instruction.substr(0, 3).str() + '}' +
+ Instruction.substr(3).str();
+ else if (Instruction.startswith("vex2 ") ||
+ Instruction.startswith("vex3 ") || Instruction.startswith("evex "))
+ MSAsmString += '{' + Instruction.substr(0, 4).str() + '}' +
+ Instruction.substr(4).str();
+ else
+ MSAsmString += Instruction.str();
+ // If this is not the last instruction, adding back the '\n\t'.
+ if (i < e - 1)
+ MSAsmString += "\n\t";
+ }
+ return MSAsmString;
}
Expr *MSAsmStmt::getOutputExpr(unsigned i) {
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
