1. Updated function headers for all assembly function

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLibMmx / Ia32 / SetMem16.asm

diff --git a/MdePkg/Library/BaseMemoryLibMmx/Ia32/SetMem16.asm b/MdePkg/Library/BaseMemoryLibMmx/Ia32/SetMem16.asm
index f9dc533fd8dceae8fdc51af100c3a4bcb57aa8eb..e93422665822a306a9482a4261937f9f7896fbe9 100644 (file)
--- a/MdePkg/Library/BaseMemoryLibMmx/Ia32/SetMem16.asm
+++ b/MdePkg/Library/BaseMemoryLibMmx/Ia32/SetMem16.asm
@@ -23,32 +23,36 @@
 \r
     .686\r
     .model  flat,C\r
-    .xmm\r
+    .mmx\r
     .code\r
 \r
 ;------------------------------------------------------------------------------\r
 ;  VOID *\r
-;  _mem_SetMem16 (\r
+;  EFIAPI\r
+;  InternalMemSetMem16 (\r
 ;    IN VOID   *Buffer,\r
 ;    IN UINTN  Count,\r
 ;    IN UINT16 Value\r
 ;    )\r
 ;------------------------------------------------------------------------------\r
 InternalMemSetMem16 PROC    USES    edi\r
+    mov     eax, [esp + 16]\r
+    shrd    edx, eax, 16\r
+    shld    eax, edx, 16\r
     mov     edx, [esp + 12]\r
     mov     edi, [esp + 8]\r
     mov     ecx, edx\r
     and     edx, 3\r
     shr     ecx, 2\r
-    mov     eax, [esp + 16]\r
     jz      @SetWords\r
     movd    mm0, eax\r
-    pshufw  mm0, mm0, 0\r
+    movd    mm1, eax\r
+    psllq   mm0, 32\r
+    por     mm0, mm1\r
 @@:\r
-    movntq  [edi], mm0\r
+    movq    [edi], mm0\r
     add     edi, 8\r
     loop    @B\r
-    mfence\r
 @SetWords:\r
     mov     ecx, edx\r
     rep     stosw\r