Update SIMD splat instructions

Signed-off-by: HyukWoo Park <[email protected]>
clover2123 · Oct 31, 2023 · b7b8316 · b7b8316
1 parent 47e0d12
commit b7b8316
Show file tree

Hide file tree

Showing 3 changed files with 446 additions and 1 deletion.
diff --git a/src/interpreter/ByteCode.h b/src/interpreter/ByteCode.h
@@ -423,12 +423,16 @@ class FunctionType;
     F(V128AnyTrue, (simdAnyTrueOperation))                                          \
     F(I8X16Bitmask, (simdBitmaskOperation<int8_t>))                                 \
     F(I8X16AllTrue, (simdAllTrueOperation<uint8_t, 16>))                            \
+    F(I8X16Splat, (simdSplatOperation<uint32_t, uint8_t>))                          \
     F(I16X8Bitmask, (simdBitmaskOperation<int16_t>))                                \
     F(I16X8AllTrue, (simdAllTrueOperation<uint16_t, 8>))                            \
+    F(I16X8Splat, (simdSplatOperation<uint32_t, uint16_t>))                         \
     F(I32X4Bitmask, (simdBitmaskOperation<int32_t>))                                \
     F(I32X4AllTrue, (simdAllTrueOperation<uint32_t, 4>))                            \
+    F(I32X4Splat, (simdSplatOperation<uint32_t, uint32_t>))                         \
     F(I64X2Bitmask, (simdBitmaskOperation<int64_t>))                                \
     F(I64X2AllTrue, (simdAllTrueOperation<uint64_t, 2>))                            \
+    F(I64X2Splat, (simdSplatOperation<uint64_t, uint64_t>))                         \
     F(I16X8ExtaddPairwiseI8X16S, (simdExtaddPairwiseOperation<int8_t, int16_t>))    \
     F(I16X8ExtaddPairwiseI8X16U, (simdExtaddPairwiseOperation<uint8_t, uint16_t>))  \
     F(I32X4ExtaddPairwiseI16X8S, (simdExtaddPairwiseOperation<int16_t, int32_t>))   \
@@ -439,7 +443,9 @@ class FunctionType;
     F(I32X4TruncSatF64X2UZero, (simdTruncSatZeroOperation<double, uint32_t>))       \
     F(F32X4ConvertI32X4S, (simdConvertOperation<int32_t, float>))                   \
     F(F32X4ConvertI32X4U, (simdConvertOperation<uint32_t, float>))                  \
-    F(F32X4DemoteF64X2Zero, (simdDemoteZeroOperation))
+    F(F32X4DemoteF64X2Zero, (simdDemoteZeroOperation))                              \
+    F(F32X4Splat, (simdSplatOperation<float, float>))                               \
+    F(F64X2Splat, (simdSplatOperation<double, double>))
 
 #define FOR_EACH_BYTECODE_SIMD_LOAD_SPLAT_OP(F) \
     F(V128Load8Splat, uint8_t)                  \

diff --git a/src/interpreter/Interpreter.cpp b/src/interpreter/Interpreter.cpp
@@ -450,6 +450,15 @@ inline static void simdDemoteZeroOperation(ExecutionState& state, UnaryOperation
     writeValue<ResultType>(bp, code->dstOffset(), result);
 }
 
+template <typename P, typename R>
+inline static void simdSplatOperation(ExecutionState& state, UnaryOperation* code, uint8_t* bp)
+{
+    using ResultType = typename SIMDType<R>::Type;
+    auto val = readValue<P>(bp, code->srcOffset());
+    ResultType result;
+    std::fill(std::begin(result.v), std::end(result.v), val);
+    writeValue<ResultType>(bp, code->dstOffset(), result);
+}
 
 #if defined(WALRUS_ENABLE_COMPUTED_GOTO)
 static void initAddressToOpcodeTable()