[vm] x64: add pshufd

Needed to improve the performance of Int32x4.shuffle and
Float32x4.shuffle.

Issue: #64170
Issue: #64345
TEST=vm/cc/Assembler_Pshufd

Change-Id: Icadaffff8e418bf90168daf428ca533f21f70da0
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/559000
Auto-Submit: Modestas Valauskas <valauskasmodestas@gmail.com>
Commit-Queue: Alexander Aprelev <aam@google.com>
Reviewed-by: Alexander Aprelev <aam@google.com>
Reviewed-by: Slava Egorov <vegorov@google.com>

Author: modulovalue

runtime/vm/compiler/assembler/assembler_x64.cc

@@ -573,6 +573,13 @@ void Assembler::shufpd(XmmRegister dst, XmmRegister src, const Immediate& imm) {
   EmitUint8(imm.value());
 }
 
+void Assembler::pshufd(XmmRegister dst, XmmRegister src, const Immediate& imm) {
+  EmitL(dst, src, 0x70, 0x0F, 0x66);
+  AssemblerBuffer::EnsureCapacity ensured(&buffer_);
+  ASSERT(imm.is_uint8());
+  EmitUint8(imm.value());
+}
+
 void Assembler::roundsd(XmmRegister dst, XmmRegister src, RoundingMode mode) {
   ASSERT(src <= XMM15);
   ASSERT(dst <= XMM15);

runtime/vm/compiler/assembler/assembler_x64.h

@@ -562,6 +562,7 @@ class Assembler : public AssemblerBase {
   void negatepd(XmmRegister dst, XmmRegister src);
   void abspd(XmmRegister dst, XmmRegister src);
   void shufpd(XmmRegister dst, XmmRegister src, const Immediate& mask);
+  void pshufd(XmmRegister dst, XmmRegister src, const Immediate& mask);
 
   enum RoundingMode {
     kRoundToNearest = 0x0,

runtime/vm/compiler/assembler/assembler_x64_test.cc

@@ -5415,6 +5415,41 @@ ASSEMBLER_TEST_RUN(Ptest, test) {
       "ret\n");
 }
 
+ASSEMBLER_TEST_GENERATE(Pshufd, assembler) {
+  static const struct ALIGN16 {
+    uint32_t a;
+    uint32_t b;
+    uint32_t c;
+    uint32_t d;
+  } constant1 = {0x01234567, 0x89ABCDEF, 0x76543210, 0xFEDCBA98};
+  __ movq(RAX, CallingConventions::kArg1Reg);
+  __ movq(RDX, Immediate(reinterpret_cast<intptr_t>(&constant1)));
+  __ movups(XMM0, Address(RDX, 0));
+  __ pshufd(XMM1, XMM0, Immediate(0x9C));
+  __ movups(Address(RAX, 0), XMM1);
+  __ movups(Address(RAX, 16), XMM0);
+  __ ret();
+}
+
+ASSEMBLER_TEST_RUN(Pshufd, test) {
+  typedef void (*PshufdCode)(uint32_t* result);
+  uint32_t result[8] = {0, 0, 0, 0, 0, 0, 0, 0};
+  reinterpret_cast<PshufdCode>(test->entry())(result);
+  EXPECT_EQ(0x01234567u, result[0]);
+  EXPECT_EQ(0xFEDCBA98u, result[1]);
+  EXPECT_EQ(0x89ABCDEFu, result[2]);
+  EXPECT_EQ(0x76543210u, result[3]);
+  EXPECT_EQ(0x01234567u, result[4]);
+  EXPECT_EQ(0x89ABCDEFu, result[5]);
+  EXPECT_EQ(0x76543210u, result[6]);
+  EXPECT_EQ(0xFEDCBA98u, result[7]);
+  EXPECT_DISASSEMBLY_ENDS_WITH(
+      "pshufd xmm1,xmm0 [9c]\n"
+      "movups [rax],xmm1\n"
+      "movups [rax+0x10],xmm0\n"
+      "ret\n");
+}
+
 ASSEMBLER_TEST_GENERATE(SquareRootDouble, assembler) {
   __ sqrtsd(XMM0, XMM0);
   __ ret();

runtime/vm/compiler/assembler/disassembler_x86.cc

@@ -1290,6 +1290,12 @@ int DisassemblerX64::TwoByteOpcodeInstruction(uint8_t* data) {
       current += PrintRightXMMOperand(current);
       Print(" [%x]", *current);
       current++;
+    } else if (opcode == 0x70) {
+      get_modrm(*current, &mod, &regop, &rm);
+      Print("pshufd %s,", NameOfXMMRegister(regop));
+      current += PrintRightXMMOperand(current);
+      Print(" [%x]", *current);
+      current++;
     } else if (opcode == 0x3A) {
       uint8_t third_byte = *current;
       current = data + 3;