Reland MIPS: Vector-ICs - speed towards the monomorphic exit as quickly as possible.

src/mips64/code-stubs-mips64.cc

 // Copyright 2012 the V8 project authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "src/v8.h"
 
 #if V8_TARGET_ARCH_MIPS64
 
 #include "src/bootstrapper.h"
 #include "src/code-stubs.h"
@@ skipping 4628 matching lines @@
 }
 
 
 void VectorRawLoadStub::GenerateForTrampoline(MacroAssembler* masm) {
   GenerateImpl(masm, true);
 }
 
 
 static void HandleArrayCases(MacroAssembler* masm, Register receiver,
                              Register key, Register vector, Register slot,
-                             Register feedback, Register scratch1,
-                             Register scratch2, Register scratch3,
+                             Register feedback, Register receiver_map,
+                             Register scratch1, Register scratch2,
                              bool is_polymorphic, Label* miss) {
   // feedback initially contains the feedback array
   Label next_loop, prepare_next;
-  Label load_smi_map, compare_map;
   Label start_polymorphic;
 
-  Register receiver_map = scratch1;
-  Register cached_map = scratch2;
+  Register cached_map = scratch1;
 
-  // Receiver might not be a heap object.
-  __ JumpIfSmi(receiver, &load_smi_map);
-  __ ld(receiver_map, FieldMemOperand(receiver, HeapObject::kMapOffset));
-  __ bind(&compare_map);
   __ ld(cached_map,
         FieldMemOperand(feedback, FixedArray::OffsetOfElementAt(0)));
   __ ld(cached_map, FieldMemOperand(cached_map, WeakCell::kValueOffset));
   __ Branch(&start_polymorphic, ne, receiver_map, Operand(cached_map));
   // found, now call handler.
   Register handler = feedback;
   __ ld(handler, FieldMemOperand(feedback, FixedArray::OffsetOfElementAt(1)));
   __ Daddu(t9, handler, Operand(Code::kHeaderSize - kHeapObjectTag));
   __ Jump(t9);
 
-  Register length = scratch3;
+  Register length = scratch2;
   __ bind(&start_polymorphic);
   __ ld(length, FieldMemOperand(feedback, FixedArray::kLengthOffset));
   if (!is_polymorphic) {
     // If the IC could be monomorphic we have to make sure we don't go past the
     // end of the feedback array.
     __ Branch(miss, eq, length, Operand(Smi::FromInt(2)));
   }
 
   Register too_far = length;
   Register pointer_reg = feedback;
 
   // +-----+------+------+-----+-----+ ... ----+
   // | map | len  | wm0  | h0  | wm1 |      hN |
   // +-----+------+------+-----+-----+ ... ----+
   //                 0      1     2        len-1
   //                              ^              ^
   //                              |              |
   //                         pointer_reg      too_far
-  //                         aka feedback     scratch3
-  // also need receiver_map (aka scratch1)
-  // use cached_map (scratch2) to look in the weak map values.
+  //                         aka feedback     scratch2
+  // also need receiver_map
+  // use cached_map (scratch1) to look in the weak map values.
   __ SmiScale(too_far, length, kPointerSizeLog2);
   __ Daddu(too_far, feedback, Operand(too_far));
   __ Daddu(too_far, too_far, Operand(FixedArray::kHeaderSize - kHeapObjectTag));
   __ Daddu(pointer_reg, feedback,
            Operand(FixedArray::OffsetOfElementAt(2) - kHeapObjectTag));
 
   __ bind(&next_loop);
   __ ld(cached_map, MemOperand(pointer_reg));
   __ ld(cached_map, FieldMemOperand(cached_map, WeakCell::kValueOffset));
   __ Branch(&prepare_next, ne, receiver_map, Operand(cached_map));
   __ ld(handler, MemOperand(pointer_reg, kPointerSize));
   __ Daddu(t9, handler, Operand(Code::kHeaderSize - kHeapObjectTag));
   __ Jump(t9);
 
   __ bind(&prepare_next);
   __ Daddu(pointer_reg, pointer_reg, Operand(kPointerSize * 2));
   __ Branch(&next_loop, lt, pointer_reg, Operand(too_far));
 
   // We exhausted our array of map handler pairs.
   __ Branch(miss);
-
-  __ bind(&load_smi_map);
-  __ LoadRoot(receiver_map, Heap::kHeapNumberMapRootIndex);
-  __ Branch(&compare_map);
 }
 
 
 static void HandleMonomorphicCase(MacroAssembler* masm, Register receiver,
-                                  Register key, Register vector, Register slot,
-                                  Register weak_cell, Register scratch,
-                                  Label* miss) {
-  // feedback initially contains the feedback array
-  Label compare_smi_map;
-  Register receiver_map = scratch;
-  Register cached_map = weak_cell;
-
+                                  Register receiver_map, Register feedback,
+                                  Register vector, Register slot,
+                                  Register scratch, Label* compare_map,
+                                  Label* load_smi_map, Label* try_array) {
+  __ JumpIfSmi(receiver, load_smi_map);
+  __ ld(receiver_map, FieldMemOperand(receiver, HeapObject::kMapOffset));
+  __ bind(compare_map);
+  Register cached_map = scratch;
   // Move the weak map into the weak_cell register.
-  __ ld(cached_map, FieldMemOperand(weak_cell, WeakCell::kValueOffset));
-
-  // Receiver might not be a heap object.
-  __ JumpIfSmi(receiver, &compare_smi_map);
-  __ ld(receiver_map, FieldMemOperand(receiver, HeapObject::kMapOffset));
-  __ Branch(miss, ne, cached_map, Operand(receiver_map));
-
-  Register handler = weak_cell;
+  __ ld(cached_map, FieldMemOperand(feedback, WeakCell::kValueOffset));
+  __ Branch(try_array, ne, cached_map, Operand(receiver_map));
+  Register handler = feedback;
   __ SmiScale(handler, slot, kPointerSizeLog2);
   __ Daddu(handler, vector, Operand(handler));
   __ ld(handler,
-        FieldMemOperand(handler, FixedArray::kHeaderSize + kPointerSize));
-  __ Daddu(t9, handler, Code::kHeaderSize - kHeapObjectTag);
-  __ Jump(t9);
-
-  // In microbenchmarks, it made sense to unroll this code so that the call to
-  // the handler is duplicated for a HeapObject receiver and a Smi receiver.
-  // TODO(mvstanton): does this hold on ARM?
-  __ bind(&compare_smi_map);
-  __ LoadRoot(at, Heap::kHeapNumberMapRootIndex);
-  __ Branch(miss, ne, weak_cell, Operand(at));
-  __ SmiScale(handler, slot, kPointerSizeLog2);
-  __ Daddu(handler, vector, Operand(handler));
-  __ ld(handler,
         FieldMemOperand(handler, FixedArray::kHeaderSize + kPointerSize));
   __ Daddu(t9, handler, Code::kHeaderSize - kHeapObjectTag);
   __ Jump(t9);
 }
 
 
 void VectorRawLoadStub::GenerateImpl(MacroAssembler* masm, bool in_frame) {
   Register receiver = VectorLoadICDescriptor::ReceiverRegister();  // a1
   Register name = VectorLoadICDescriptor::NameRegister();          // a2
   Register vector = VectorLoadICDescriptor::VectorRegister();      // a3
   Register slot = VectorLoadICDescriptor::SlotRegister();          // a0
   Register feedback = a4;
-  Register scratch1 = a5;
+  Register receiver_map = a5;
+  Register scratch1 = a6;
 
   __ SmiScale(feedback, slot, kPointerSizeLog2);
   __ Daddu(feedback, vector, Operand(feedback));
   __ ld(feedback, FieldMemOperand(feedback, FixedArray::kHeaderSize));
 
-  // Is it a weak cell?
-  Label try_array;
-  Label not_array, smi_key, key_okay, miss;
-  __ ld(scratch1, FieldMemOperand(feedback, HeapObject::kMapOffset));
-  __ LoadRoot(at, Heap::kWeakCellMapRootIndex);
-  __ Branch(&try_array, ne, scratch1, Operand(at));
-  HandleMonomorphicCase(masm, receiver, name, vector, slot, feedback, scratch1,
-                        &miss);
+  // Try to quickly handle the monomorphic case without knowing for sure
+  // if we have a weak cell in feedback. We do know it's safe to look
+  // at WeakCell::kValueOffset.
+  Label try_array, load_smi_map, compare_map;
+  Label not_array, miss;
+  HandleMonomorphicCase(masm, receiver, receiver_map, feedback, vector, slot,
+                        scratch1, &compare_map, &load_smi_map, &try_array);
 
   // Is it a fixed array?
   __ bind(&try_array);
+  __ ld(scratch1, FieldMemOperand(feedback, HeapObject::kMapOffset));
   __ LoadRoot(at, Heap::kFixedArrayMapRootIndex);
   __ Branch(&not_array, ne, scratch1, Operand(at));
-  HandleArrayCases(masm, receiver, name, vector, slot, feedback, scratch1, a6,
-                   a7, true, &miss);
+  HandleArrayCases(masm, receiver, name, vector, slot, feedback, receiver_map,
+                   scratch1, a7, true, &miss);
 
   __ bind(&not_array);
   __ LoadRoot(at, Heap::kmegamorphic_symbolRootIndex);
   __ Branch(&miss, ne, feedback, Operand(at));
   Code::Flags code_flags = Code::RemoveTypeAndHolderFromFlags(
       Code::ComputeHandlerFlags(Code::LOAD_IC));
   masm->isolate()->stub_cache()->GenerateProbe(masm, Code::LOAD_IC, code_flags,
                                                false, receiver, name, feedback,
-                                               scratch1, a6, a7);
+                                               receiver_map, scratch1, a7);
 
   __ bind(&miss);
   LoadIC::GenerateMiss(masm);
+
+  __ bind(&load_smi_map);
+  __ LoadRoot(receiver_map, Heap::kHeapNumberMapRootIndex);
+  __ Branch(&compare_map);
 }
 
 
 void VectorRawKeyedLoadStub::Generate(MacroAssembler* masm) {
   GenerateImpl(masm, false);
 }
 
 
 void VectorRawKeyedLoadStub::GenerateForTrampoline(MacroAssembler* masm) {
   GenerateImpl(masm, true);
 }
 
 
 void VectorRawKeyedLoadStub::GenerateImpl(MacroAssembler* masm, bool in_frame) {
   Register receiver = VectorLoadICDescriptor::ReceiverRegister();  // a1
   Register key = VectorLoadICDescriptor::NameRegister();           // a2
   Register vector = VectorLoadICDescriptor::VectorRegister();      // a3
   Register slot = VectorLoadICDescriptor::SlotRegister();          // a0
   Register feedback = a4;
-  Register scratch1 = a5;
+  Register receiver_map = a5;
+  Register scratch1 = a6;
 
   __ SmiScale(feedback, slot, kPointerSizeLog2);
   __ Daddu(feedback, vector, Operand(feedback));
   __ ld(feedback, FieldMemOperand(feedback, FixedArray::kHeaderSize));
 
-  // Is it a weak cell?
-  Label try_array;
-  Label not_array, smi_key, key_okay, miss;
-  __ ld(scratch1, FieldMemOperand(feedback, HeapObject::kMapOffset));
-  __ LoadRoot(at, Heap::kWeakCellMapRootIndex);
-  __ Branch(&try_array, ne, scratch1, Operand(at));
-  __ JumpIfNotSmi(key, &miss);
-  HandleMonomorphicCase(masm, receiver, key, vector, slot, feedback, scratch1,
-                        &miss);
+  // Try to quickly handle the monomorphic case without knowing for sure
+  // if we have a weak cell in feedback. We do know it's safe to look
+  // at WeakCell::kValueOffset.
+  Label try_array, load_smi_map, compare_map;
+  Label not_array, miss;
+  HandleMonomorphicCase(masm, receiver, receiver_map, feedback, vector, slot,
+                        scratch1, &compare_map, &load_smi_map, &try_array);
 
   __ bind(&try_array);
   // Is it a fixed array?
+  __ ld(scratch1, FieldMemOperand(feedback, HeapObject::kMapOffset));
   __ LoadRoot(at, Heap::kFixedArrayMapRootIndex);
   __ Branch(&not_array, ne, scratch1, Operand(at));
   // We have a polymorphic element handler.
   __ JumpIfNotSmi(key, &miss);
 
   Label polymorphic, try_poly_name;
   __ bind(&polymorphic);
-  HandleArrayCases(masm, receiver, key, vector, slot, feedback, scratch1, a6,
-                   a7, true, &miss);
+  HandleArrayCases(masm, receiver, key, vector, slot, feedback, receiver_map,
+                   scratch1, a7, true, &miss);
 
   __ bind(&not_array);
   // Is it generic?
   __ LoadRoot(at, Heap::kmegamorphic_symbolRootIndex);
   __ Branch(&try_poly_name, ne, feedback, Operand(at));
   Handle<Code> megamorphic_stub =
       KeyedLoadIC::ChooseMegamorphicStub(masm->isolate());
   __ Jump(megamorphic_stub, RelocInfo::CODE_TARGET);
 
   __ bind(&try_poly_name);
   // We might have a name in feedback, and a fixed array in the next slot.
   __ Branch(&miss, ne, key, Operand(feedback));
   // If the name comparison succeeded, we know we have a fixed array with
   // at least one map/handler pair.
   __ SmiScale(feedback, slot, kPointerSizeLog2);
   __ Daddu(feedback, vector, Operand(feedback));
   __ ld(feedback,
         FieldMemOperand(feedback, FixedArray::kHeaderSize + kPointerSize));
-  HandleArrayCases(masm, receiver, key, vector, slot, feedback, scratch1, a6,
-                   a7, false, &miss);
+  HandleArrayCases(masm, receiver, key, vector, slot, feedback, receiver_map,
+                   scratch1, a7, false, &miss);
 
   __ bind(&miss);
   KeyedLoadIC::GenerateMiss(masm);
+
+  __ bind(&load_smi_map);
+  __ LoadRoot(receiver_map, Heap::kHeapNumberMapRootIndex);
+  __ Branch(&compare_map);
 }
 
 
 void ProfileEntryHookStub::MaybeCallEntryHook(MacroAssembler* masm) {
   if (masm->isolate()->function_entry_hook() != NULL) {
     ProfileEntryHookStub stub(masm->isolate());
     __ push(ra);
     __ CallStub(&stub);
     __ pop(ra);
   }
@@ skipping 675 matching lines @@
                            kStackUnwindSpace, kInvalidStackOffset,
                            MemOperand(fp, 6 * kPointerSize), NULL);
 }
 
 
 #undef __
 
 } }  // namespace v8::internal
 
 #endif  // V8_TARGET_ARCH_MIPS64