[Atomics] Make Atomics.exchange a builtin using TF

src/builtins/builtins-sharedarraybuffer.cc

 // Copyright 2016 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/builtins/builtins-utils.h"
 #include "src/builtins/builtins.h"
 #include "src/code-factory.h"
 #include "src/code-stub-assembler.h"
 
 namespace v8 {
@@ skipping 75 matching lines @@
   Node* backing_store =
       a->LoadObjectField(array_buffer, JSArrayBuffer::kBackingStoreOffset);
   Node* byte_offset = a->ChangeUint32ToWord(a->TruncateTaggedToWord32(
       context,
       a->LoadObjectField(tagged, JSArrayBufferView::kByteOffsetOffset)));
   *out_backing_store =
       a->IntPtrAdd(a->BitcastTaggedToWord(backing_store), byte_offset);
 }
 
 // https://tc39.github.io/ecmascript_sharedmem/shmem.html#Atomics.ValidateAtomicAccess
-compiler::Node* ConvertTaggedAtomicIndexToWord32(CodeStubAssembler* a,
-                                                 compiler::Node* tagged,
-                                                 compiler::Node* context) {
+compiler::Node* ConvertTaggedAtomicIndexToWord32(
+    CodeStubAssembler* a, compiler::Node* tagged, compiler::Node* context,
+    compiler::Node** number_index) {
   using compiler::Node;
   CodeStubAssembler::Variable var_result(a, MachineRepresentation::kWord32);
 
   Callable to_number = CodeFactory::ToNumber(a->isolate());
-  Node* number_index = a->CallStub(to_number, context, tagged);
+  *number_index = a->CallStub(to_number, context, tagged);
   CodeStubAssembler::Label done(a, &var_result);
 
   CodeStubAssembler::Label if_numberissmi(a), if_numberisnotsmi(a);
-  a->Branch(a->TaggedIsSmi(number_index), &if_numberissmi, &if_numberisnotsmi);
+  a->Branch(a->TaggedIsSmi(*number_index), &if_numberissmi, &if_numberisnotsmi);
 
   a->Bind(&if_numberissmi);
   {
-    var_result.Bind(a->SmiToWord32(number_index));
+    var_result.Bind(a->SmiToWord32(*number_index));
     a->Goto(&done);
   }
 
   a->Bind(&if_numberisnotsmi);
   {
-    Node* number_index_value = a->LoadHeapNumberValue(number_index);
+    Node* number_index_value = a->LoadHeapNumberValue(*number_index);
     Node* access_index = a->TruncateFloat64ToWord32(number_index_value);
     Node* test_index = a->ChangeInt32ToFloat64(access_index);
 
     CodeStubAssembler::Label if_indexesareequal(a), if_indexesarenotequal(a);
     a->Branch(a->Float64Equal(number_index_value, test_index),
               &if_indexesareequal, &if_indexesarenotequal);
 
     a->Bind(&if_indexesareequal);
     {
       var_result.Bind(access_index);
@@ skipping 32 matching lines @@
   using compiler::Node;
   CodeStubAssembler a(state);
   Node* array = a.Parameter(1);
   Node* index = a.Parameter(2);
   Node* context = a.Parameter(3 + 2);
 
   Node* instance_type;
   Node* backing_store;
   ValidateSharedTypedArray(&a, array, context, &instance_type, &backing_store);
 
-  Node* index_word32 = ConvertTaggedAtomicIndexToWord32(&a, index, context);
+  Node* index_integer;
+  Node* index_word32 =
+      ConvertTaggedAtomicIndexToWord32(&a, index, context, &index_integer);
   Node* array_length_word32 = a.TruncateTaggedToWord32(
       context, a.LoadObjectField(array, JSTypedArray::kLengthOffset));
   ValidateAtomicIndex(&a, index_word32, array_length_word32, context);
   Node* index_word = a.ChangeUint32ToWord(index_word32);
 
   CodeStubAssembler::Label i8(&a), u8(&a), i16(&a), u16(&a), i32(&a), u32(&a),
       other(&a);
   int32_t case_values[] = {
       FIXED_INT8_ARRAY_TYPE,   FIXED_UINT8_ARRAY_TYPE, FIXED_INT16_ARRAY_TYPE,
       FIXED_UINT16_ARRAY_TYPE, FIXED_INT32_ARRAY_TYPE, FIXED_UINT32_ARRAY_TYPE,
@@ skipping 38 matching lines @@
   CodeStubAssembler a(state);
   Node* array = a.Parameter(1);
   Node* index = a.Parameter(2);
   Node* value = a.Parameter(3);
   Node* context = a.Parameter(4 + 2);
 
   Node* instance_type;
   Node* backing_store;
   ValidateSharedTypedArray(&a, array, context, &instance_type, &backing_store);
 
-  Node* index_word32 = ConvertTaggedAtomicIndexToWord32(&a, index, context);
+  Node* index_integer;
+  Node* index_word32 =
+      ConvertTaggedAtomicIndexToWord32(&a, index, context, &index_integer);
   Node* array_length_word32 = a.TruncateTaggedToWord32(
       context, a.LoadObjectField(array, JSTypedArray::kLengthOffset));
   ValidateAtomicIndex(&a, index_word32, array_length_word32, context);
   Node* index_word = a.ChangeUint32ToWord(index_word32);
 
   Node* value_integer = a.ToInteger(context, value);
   Node* value_word32 = a.TruncateTaggedToWord32(context, value_integer);
 
   CodeStubAssembler::Label u8(&a), u16(&a), u32(&a), other(&a);
   int32_t case_values[] = {
@@ skipping 19 matching lines @@
   a.Bind(&u32);
   a.AtomicStore(MachineRepresentation::kWord32, backing_store,
                 a.WordShl(index_word, 2), value_word32);
   a.Return(value_integer);
 
   // This shouldn't happen, we've already validated the type.
   a.Bind(&other);
   a.Return(a.SmiConstant(0));
 }
 
+void Builtins::Generate_AtomicsExchange(compiler::CodeAssemblerState* state) {
+  using compiler::Node;
+  CodeStubAssembler a(state);
+  Node* array = a.Parameter(1);
+  Node* index = a.Parameter(2);
+  Node* value = a.Parameter(3);
+  Node* context = a.Parameter(4 + 2);
+
+  Node* instance_type;
+  Node* backing_store;
+  ValidateSharedTypedArray(&a, array, context, &instance_type, &backing_store);
+
+  Node* index_integer;
+  Node* index_word32 =
+      ConvertTaggedAtomicIndexToWord32(&a, index, context, &index_integer);
+  Node* array_length_word32 = a.TruncateTaggedToWord32(
+      context, a.LoadObjectField(array, JSTypedArray::kLengthOffset));
+  ValidateAtomicIndex(&a, index_word32, array_length_word32, context);
+
+  Node* value_integer = a.ToInteger(context, value);
+
+#if V8_TARGET_ARCH_MIPS || V8_TARGET_ARCH_MIPS64
+  // Node* index_integer = a.ToInteger(context, index);
+  a.Return(a.CallRuntime(Runtime::kAtomicsExchange, context, array,
+                         index_integer, value_integer));

[Jarin, 2017/03/06 10:59:05]

Can't you tail-call to the runtime?

I was hoping you could say here "a.TailCallRuntime(..." instead of
"a.Return(a.CallRuntime".

[aseemgarg, 2017/03/06 23:20:07]

I tried doing a.TailCallRuntime instead of a.Return(a.CallRuntime(.. However, I
keep getting an error "abort: The stack pointer is not the expected value" I
can't figure out why that's happening, so I figured would go with this since it
is supposed to be removed eventually anyways.

+#else
+  Node* index_word = a.ChangeUint32ToWord(index_word32);
+
+  Node* value_word32 = a.TruncateTaggedToWord32(context, value_integer);
+
+  CodeStubAssembler::Label i8(&a), u8(&a), i16(&a), u16(&a), i32(&a), u32(&a),
+      other(&a);
+  int32_t case_values[] = {
+      FIXED_INT8_ARRAY_TYPE,   FIXED_UINT8_ARRAY_TYPE, FIXED_INT16_ARRAY_TYPE,
+      FIXED_UINT16_ARRAY_TYPE, FIXED_INT32_ARRAY_TYPE, FIXED_UINT32_ARRAY_TYPE,
+  };
+  CodeStubAssembler::Label* case_labels[] = {
+      &i8, &u8, &i16, &u16, &i32, &u32,
+  };
+  a.Switch(instance_type, &other, case_values, case_labels,
+           arraysize(case_labels));
+
+  a.Bind(&i8);
+  a.Return(a.SmiFromWord32(a.AtomicExchange(MachineType::Int8(), backing_store,
+                                            index_word, value_word32)));
+
+  a.Bind(&u8);
+  a.Return(a.SmiFromWord32(a.AtomicExchange(MachineType::Uint8(), backing_store,
+                                            index_word, value_word32)));
+
+  a.Bind(&i16);
+  a.Return(a.SmiFromWord32(a.AtomicExchange(MachineType::Int16(), backing_store,
+                                            a.WordShl(index_word, 1),
+                                            value_word32)));
+
+  a.Bind(&u16);
+  a.Return(a.SmiFromWord32(
+      a.AtomicExchange(MachineType::Uint16(), backing_store,
+                       a.WordShl(index_word, 1), value_word32)));
+
+  a.Bind(&i32);
+  a.Return(a.ChangeInt32ToTagged(
+      a.AtomicExchange(MachineType::Int32(), backing_store,
+                       a.WordShl(index_word, 2), value_word32)));
+
+  a.Bind(&u32);
+  a.Return(a.ChangeUint32ToTagged(
+      a.AtomicExchange(MachineType::Uint32(), backing_store,
+                       a.WordShl(index_word, 2), value_word32)));
+
+  // This shouldn't happen, we've already validated the type.
+  a.Bind(&other);
+  a.Return(a.SmiConstant(0));
+#endif  // V8_TARGET_ARCH_MIPS || V8_TARGET_ARCH_MIPS64
+}
+
 }  // namespace internal
 }  // namespace v8