SIMD.js Add the other SIMD Phase 1 types.

src/runtime/runtime-simd.cc

 // Copyright 2015 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"
 
 #include "src/arguments.h"
 #include "src/base/macros.h"
 #include "src/conversions.h"
 #include "src/runtime/runtime-utils.h"
 
 // Implement Single Instruction Multiple Data (SIMD) operations as defined in
 // the SIMD.js draft spec:
 // http://littledan.github.io/simd.html
 
-#define NumberToFloat32x4Component NumberToFloat
-
 #define CONVERT_SIMD_LANE_ARG_CHECKED(name, index, lanes) \
-  RUNTIME_ASSERT(args[index]->IsSmi());                   \
-  int name = args.smi_at(index);                          \
+  CONVERT_INT32_ARG_CHECKED(name, index);                 \
   RUNTIME_ASSERT(name >= 0 && name < lanes);
 
-#define SIMD4_CREATE_FUNCTION(type)                                    \
-  RUNTIME_FUNCTION(Runtime_Create##type) {                             \
-    HandleScope scope(isolate);                                        \
-    DCHECK(args.length() == 4);                                        \
-    CONVERT_NUMBER_ARG_HANDLE_CHECKED(w, 0);                           \
-    CONVERT_NUMBER_ARG_HANDLE_CHECKED(x, 1);                           \
-    CONVERT_NUMBER_ARG_HANDLE_CHECKED(y, 2);                           \
-    CONVERT_NUMBER_ARG_HANDLE_CHECKED(z, 3);                           \
-    return *isolate->factory()->NewFloat32x4(                          \
-        NumberTo##type##Component(*w), NumberTo##type##Component(*x),  \
-        NumberTo##type##Component(*y), NumberTo##type##Component(*z)); \
+#define SIMD_CREATE_NUMERIC_FUNCTION(type, lane_type, lane_count) \
+  RUNTIME_FUNCTION(Runtime_Create##type) {                        \
+    HandleScope scope(isolate);                                   \
+    DCHECK(args.length() == lane_count);                          \
+    lane_type lanes[lane_count];                                  \
+    for (int i = 0; i < lane_count; i++) {                        \
+      CONVERT_NUMBER_ARG_HANDLE_CHECKED(number, i);               \
+      lanes[i] = ConvertNumber<lane_type>(number->Number());      \
+    }                                                             \
+    return *isolate->factory()->New##type(lanes);                 \
+  }
+
+#define SIMD_CREATE_BOOLEAN_FUNCTION(type, lane_count) \
+  RUNTIME_FUNCTION(Runtime_Create##type) {             \
+    HandleScope scope(isolate);                        \
+    DCHECK(args.length() == lane_count);               \
+    bool lanes[lane_count];                            \
+    for (int i = 0; i < lane_count; i++) {             \
+      lanes[i] = args[i]->BooleanValue();              \
+    }                                                  \
+    return *isolate->factory()->New##type(lanes);      \
   }
 
 #define SIMD_CREATE_WRAPPER_FUNCTION(type)                      \
   RUNTIME_FUNCTION(Runtime_New##type##Wrapper) {                \
     HandleScope scope(isolate);                                 \
     DCHECK(args.length() == 1);                                 \
     CONVERT_ARG_HANDLE_CHECKED(type, value, 0);                 \
     return *Object::ToObject(isolate, value).ToHandleChecked(); \
   }
 
 #define SIMD_CHECK_FUNCTION(type)           \
   RUNTIME_FUNCTION(Runtime_##type##Check) { \
     HandleScope scope(isolate);             \
     CONVERT_ARG_HANDLE_CHECKED(type, a, 0); \
     return *a;                              \
   }
 
-#define SIMD_EXTRACT_LANE_FUNCTION(type, lanes)               \
-  RUNTIME_FUNCTION(Runtime_##type##ExtractLane) {             \
-    HandleScope scope(isolate);                               \
-    DCHECK(args.length() == 2);                               \
-    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);                   \
-    CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, lanes);            \
-    return *isolate->factory()->NewNumber(a->get_lane(lane)); \
+#define SIMD_EXTRACT_LANE_FUNCTION(type, lanes, extract_fn)    \
+  RUNTIME_FUNCTION(Runtime_##type##ExtractLane) {              \
+    HandleScope scope(isolate);                                \
+    DCHECK(args.length() == 2);                                \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);                    \
+    CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, lanes);             \
+    return *isolate->factory()->extract_fn(a->get_lane(lane)); \
   }
 
-#define SIMD4_EQUALS_FUNCTION(type)                          \
-  RUNTIME_FUNCTION(Runtime_##type##Equals) {                 \
-    HandleScope scope(isolate);                              \
-    DCHECK(args.length() == 2);                              \
-    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);                  \
-    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);                  \
-    return Equals(a->get_lane(0), b->get_lane(0)) &&         \
-                   Equals(a->get_lane(1), b->get_lane(1)) && \
-                   Equals(a->get_lane(2), b->get_lane(2)) && \
-                   Equals(a->get_lane(3), b->get_lane(3))    \
-               ? Smi::FromInt(EQUAL)                         \
-               : Smi::FromInt(NOT_EQUAL);                    \
+#define SIMD_REPLACE_NUMERIC_LANE_FUNCTION(type, lane_type, lane_count) \
+  RUNTIME_FUNCTION(Runtime_##type##ReplaceLane) {                       \
+    HandleScope scope(isolate);                                         \
+    DCHECK(args.length() == 3);                                         \
+    CONVERT_ARG_HANDLE_CHECKED(type, simd, 0);                          \
+    CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, lane_count);                 \
+    CONVERT_NUMBER_ARG_HANDLE_CHECKED(number, 2);                       \
+    lane_type lanes[lane_count];                                        \
+    for (int i = 0; i < lane_count; i++) {                              \
+      lanes[i] = simd->get_lane(i);                                     \
+    }                                                                   \
+    lanes[lane] = ConvertNumber<lane_type>(number->Number());           \
+    Handle<type> result = isolate->factory()->New##type(lanes);         \
+    return *result;                                                     \
   }
 
-#define SIMD4_SAME_VALUE_FUNCTION(type)              \
-  RUNTIME_FUNCTION(Runtime_##type##SameValue) {      \
-    HandleScope scope(isolate);                      \
-    DCHECK(args.length() == 2);                      \
-    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);          \
-    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);          \
-    return isolate->heap()->ToBoolean(               \
-        SameValue(a->get_lane(0), b->get_lane(0)) && \
-        SameValue(a->get_lane(1), b->get_lane(1)) && \
-        SameValue(a->get_lane(2), b->get_lane(2)) && \
-        SameValue(a->get_lane(3), b->get_lane(3)));  \
+#define SIMD_REPLACE_BOOLEAN_LANE_FUNCTION(type, lane_count)    \
+  RUNTIME_FUNCTION(Runtime_##type##ReplaceLane) {               \
+    HandleScope scope(isolate);                                 \
+    DCHECK(args.length() == 3);                                 \
+    CONVERT_ARG_HANDLE_CHECKED(type, simd, 0);                  \
+    CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, lane_count);         \
+    bool lanes[lane_count];                                     \
+    for (int i = 0; i < lane_count; i++) {                      \
+      lanes[i] = simd->get_lane(i);                             \
+    }                                                           \
+    lanes[lane] = args[2]->BooleanValue();                      \
+    Handle<type> result = isolate->factory()->New##type(lanes); \
+    return *result;                                             \
   }
 
-#define SIMD4_SAME_VALUE_ZERO_FUNCTION(type)             \
-  RUNTIME_FUNCTION(Runtime_##type##SameValueZero) {      \
-    HandleScope scope(isolate);                          \
-    DCHECK(args.length() == 2);                          \
-    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);              \
-    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);              \
-    return isolate->heap()->ToBoolean(                   \
-        SameValueZero(a->get_lane(0), b->get_lane(0)) && \
-        SameValueZero(a->get_lane(1), b->get_lane(1)) && \
-        SameValueZero(a->get_lane(2), b->get_lane(2)) && \
-        SameValueZero(a->get_lane(3), b->get_lane(3)));  \
+#define SIMD_EQUALS_FUNCTION(type, lanes)                                   \
+  RUNTIME_FUNCTION(Runtime_##type##Equals) {                                \
+    HandleScope scope(isolate);                                             \
+    DCHECK(args.length() == 2);                                             \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);                                 \
+    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);                                 \
+    for (int i = 0; i < lanes; i++) {                                       \
+      if (a->get_lane(i) != b->get_lane(i)) return Smi::FromInt(NOT_EQUAL); \
+    }                                                                       \
+    return Smi::FromInt(EQUAL);                                             \
   }
 
-#define SIMD4_EXTRACT_LANE_FUNCTION(type) SIMD_EXTRACT_LANE_FUNCTION(type, 4)
+#define SIMD_SAME_VALUE_FLOAT_FUNCTION(type, lanes)      \
+  RUNTIME_FUNCTION(Runtime_##type##SameValue) {    \
+    HandleScope scope(isolate);                    \
+    DCHECK(args.length() == 2);                    \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);        \
+    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);        \
+    bool same_value = a->SameValue(*b);            \
+    return isolate->heap()->ToBoolean(same_value); \
+  }
 
-#define SIMD4_FUNCTIONS(type)        \
-  SIMD4_CREATE_FUNCTION(type)        \
-  SIMD_CREATE_WRAPPER_FUNCTION(type) \
-  SIMD_CHECK_FUNCTION(type)          \
-  SIMD4_EXTRACT_LANE_FUNCTION(type)  \
-  SIMD4_EQUALS_FUNCTION(type)        \
-  SIMD4_SAME_VALUE_FUNCTION(type)    \
-  SIMD4_SAME_VALUE_ZERO_FUNCTION(type)
+#define SIMD_SAME_VALUE_ZERO_FLOAT_FUNCTION(type, lanes)  \
+  RUNTIME_FUNCTION(Runtime_##type##SameValueZero) { \
+    HandleScope scope(isolate);                     \
+    DCHECK(args.length() == 2);                     \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);         \
+    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);         \
+    bool same_value = a->SameValueZero(*b);         \
+    return isolate->heap()->ToBoolean(same_value);  \
+  }
+
+#define SIMD_SAME_VALUE_FUNCTION(type, lanes)      \
+  RUNTIME_FUNCTION(Runtime_##type##SameValue) {    \
+    HandleScope scope(isolate);                    \
+    DCHECK(args.length() == 2);                    \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);        \
+    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);        \
+    bool same_value = a->BitwiseEquals(*b);        \
+    return isolate->heap()->ToBoolean(same_value); \
+  }
+
+#define SIMD_SAME_VALUE_ZERO_FUNCTION(type, lanes)  \
+  RUNTIME_FUNCTION(Runtime_##type##SameValueZero) { \
+    HandleScope scope(isolate);                     \
+    DCHECK(args.length() == 2);                     \
+    CONVERT_ARG_HANDLE_CHECKED(type, a, 0);         \
+    CONVERT_ARG_HANDLE_CHECKED(type, b, 1);         \
+    bool same_value = a->BitwiseEquals(*b);         \
+    return isolate->heap()->ToBoolean(same_value);  \
+  }
+
+#define SIMD_FUNCTIONS(type, lanes, extract_fn)       \
+  SIMD_CREATE_WRAPPER_FUNCTION(type)                  \
+  SIMD_CHECK_FUNCTION(type)                           \
+  SIMD_EXTRACT_LANE_FUNCTION(type, lanes, extract_fn) \
+  SIMD_EQUALS_FUNCTION(type, lanes)
 
 
 namespace v8 {
 namespace internal {
 
 namespace {
 
-// Convert from Number object to float.
-inline float NumberToFloat(Object* number) {
-  return DoubleToFloat32(number->Number());
+// Functions to convert Numbers to SIMD component types.
+
+template <typename T>
+static T ConvertNumber(double number);
+
+
+template <>
+float ConvertNumber<float>(double number) {
+  return DoubleToFloat32(number);
 }
 
 
-inline bool Equals(float x, float y) { return x == y; }
+template <>
+int32_t ConvertNumber<int32_t>(double number) {
+  return DoubleToInt32(number);
+}
+
+
+template <>
+int16_t ConvertNumber<int16_t>(double number) {
+  return static_cast<int16_t>(DoubleToInt32(number));
+}
+
+
+template <>
+int8_t ConvertNumber<int8_t>(double number) {
+  return static_cast<int8_t>(DoubleToInt32(number));
+}
 
 }  // namespace
 
-SIMD4_FUNCTIONS(Float32x4)
+
+SIMD_CREATE_NUMERIC_FUNCTION(Float32x4, float, 4)
+SIMD_CREATE_NUMERIC_FUNCTION(Int32x4, int32_t, 4)
+SIMD_CREATE_BOOLEAN_FUNCTION(Bool32x4, 4)
+SIMD_CREATE_NUMERIC_FUNCTION(Int16x8, int16_t, 8)
+SIMD_CREATE_BOOLEAN_FUNCTION(Bool16x8, 8)
+SIMD_CREATE_NUMERIC_FUNCTION(Int8x16, int8_t, 16)
+SIMD_CREATE_BOOLEAN_FUNCTION(Bool8x16, 16)
+
+
+RUNTIME_FUNCTION(Runtime_Int16x8UnsignedExtractLane) {
+  HandleScope scope(isolate);
+  DCHECK(args.length() == 2);
+  CONVERT_ARG_HANDLE_CHECKED(Int16x8, a, 0);
+  CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, 8);
+  return *isolate->factory()->NewNumber(bit_cast<uint16_t>(a->get_lane(lane)));
+}
+
+
+RUNTIME_FUNCTION(Runtime_Int8x16UnsignedExtractLane) {
+  HandleScope scope(isolate);
+  DCHECK(args.length() == 2);
+  CONVERT_ARG_HANDLE_CHECKED(Int8x16, a, 0);
+  CONVERT_SIMD_LANE_ARG_CHECKED(lane, 1, 16);
+  return *isolate->factory()->NewNumber(bit_cast<uint8_t>(a->get_lane(lane)));
+}
+
+SIMD_REPLACE_NUMERIC_LANE_FUNCTION(Float32x4, float, 4)
+SIMD_REPLACE_NUMERIC_LANE_FUNCTION(Int32x4, int32_t, 4)
+SIMD_REPLACE_BOOLEAN_LANE_FUNCTION(Bool32x4, 4)
+SIMD_REPLACE_NUMERIC_LANE_FUNCTION(Int16x8, int16_t, 8)
+SIMD_REPLACE_BOOLEAN_LANE_FUNCTION(Bool16x8, 8)
+SIMD_REPLACE_NUMERIC_LANE_FUNCTION(Int8x16, int8_t, 16)
+SIMD_REPLACE_BOOLEAN_LANE_FUNCTION(Bool8x16, 16)
+
+
+SIMD_SAME_VALUE_FLOAT_FUNCTION(Float32x4, 4)
+SIMD_SAME_VALUE_FUNCTION(Int32x4, 4)
+SIMD_SAME_VALUE_FUNCTION(Bool32x4, 4)
+SIMD_SAME_VALUE_FUNCTION(Int16x8, 8)
+SIMD_SAME_VALUE_FUNCTION(Bool16x8, 8)
+SIMD_SAME_VALUE_FUNCTION(Int8x16, 16)
+SIMD_SAME_VALUE_FUNCTION(Bool8x16, 16)
+
+
+SIMD_SAME_VALUE_ZERO_FLOAT_FUNCTION(Float32x4, 4)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Int32x4, 4)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Bool32x4, 4)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Int16x8, 8)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Bool16x8, 8)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Int8x16, 16)
+SIMD_SAME_VALUE_ZERO_FUNCTION(Bool8x16, 16)
+
+
+SIMD_FUNCTIONS(Float32x4, 4, NewNumber)
+SIMD_FUNCTIONS(Int32x4, 4, NewNumber)
+SIMD_FUNCTIONS(Bool32x4, 4, ToBoolean)
+SIMD_FUNCTIONS(Int16x8, 8, NewNumber)
+SIMD_FUNCTIONS(Bool16x8, 8, ToBoolean)
+SIMD_FUNCTIONS(Int8x16, 16, NewNumber)
+SIMD_FUNCTIONS(Bool8x16, 16, ToBoolean)
 }
 }  // namespace v8::internal