[Embedded] Enable SIMD types in embedded Swift

Author: drexin

stdlib/public/core/CMakeLists.txt

@@ -252,6 +252,9 @@ split_embedded_sources(
 if(SWIFT_STDLIB_ENABLE_VECTOR_TYPES)
   list(APPEND SWIFTLIB_SOURCES SIMDVector.swift)
   list(APPEND SWIFTLIB_GYB_SOURCES SIMDConcreteOperations.swift.gyb SIMDVectorTypes.swift.gyb)
+
+  list(APPEND SWIFTLIB_EMBEDDED_SOURCES SIMDVector.swift)
+  list(APPEND SWIFTLIB_EMBEDDED_GYB_SOURCES SIMDConcreteOperations.swift.gyb SIMDVectorTypes.swift.gyb)
 endif()
 
 list(APPEND SWIFTLIB_EMBEDDED_SOURCES

stdlib/public/core/SIMDVector.swift

@@ -33,8 +33,12 @@ prefix operator .!
 /// conform to `SIMD`.
 public protocol SIMDStorage {
   /// The type of scalars in the vector space.
+  #if $Embedded
+  associatedtype Scalar: Hashable
+  #else
   associatedtype Scalar: Codable, Hashable
-  
+  #endif
+
   /// The number of scalars, or elements, in the vector.
   var scalarCount: Int { get }
 
@@ -71,6 +75,20 @@ public protocol SIMDScalar {
   associatedtype SIMD64Storage: SIMDStorage where SIMD64Storage.Scalar == Self
 }
 
+#if $Embedded
+/// A SIMD vector of a fixed number of elements.
+public protocol SIMD<Scalar>:
+  SIMDStorage,
+  Hashable,
+  ExpressibleByArrayLiteral
+{
+  /// The mask type resulting from pointwise comparisons of this vector type.
+  associatedtype MaskStorage: SIMD
+    where MaskStorage.Scalar: FixedWidthInteger & SignedInteger
+}
+
+#else
+
 /// A SIMD vector of a fixed number of elements.
 public protocol SIMD<Scalar>:
   SIMDStorage,
@@ -84,6 +102,8 @@ public protocol SIMD<Scalar>:
     where MaskStorage.Scalar: FixedWidthInteger & SignedInteger
 }
 
+#endif
+
 extension SIMD {
   /// The valid indices for subscripting the vector.
   @_transparent
@@ -111,6 +131,8 @@ extension SIMD {
   public func hash(into hasher: inout Hasher) {
     for i in indices { hasher.combine(self[i]) }
   }
+
+#if !$Embedded
 
   /// Encodes the scalars of this vector into the given encoder in an unkeyed
   /// container.
@@ -154,7 +176,9 @@ extension SIMD {
       return "\(Self.self)(" + indices.map({"\(self[$0])"}).joined(separator: ", ") + ")"
     }
   }
-  
+
+#endif
+
   /// A vector mask with the result of a pointwise equality comparison.
   ///
   /// Equivalent to:
@@ -539,6 +563,8 @@ extension SIMD where Scalar: FixedWidthInteger {
     return Self(repeating: 1)
   }
 
+#if !$Embedded
+
   /// Returns a vector with random values from within the specified range in
   /// all lanes, using the given generator as a source for randomness.
   @inlinable
@@ -560,7 +586,7 @@ extension SIMD where Scalar: FixedWidthInteger {
     var g = SystemRandomNumberGenerator()
     return Self.random(in: range, using: &g)
   }
-  
+
   /// Returns a vector with random values from within the specified range in
   /// all lanes, using the given generator as a source for randomness.
   @inlinable
@@ -582,6 +608,9 @@ extension SIMD where Scalar: FixedWidthInteger {
     var g = SystemRandomNumberGenerator()
     return Self.random(in: range, using: &g)
   }
+
+#endif
+
 }
 
 extension SIMD where Scalar: FloatingPoint {
@@ -608,6 +637,8 @@ extension SIMD where Scalar: FloatingPoint {
   }
 }
 
+#if !$Embedded
+
 extension SIMD
 where Scalar: BinaryFloatingPoint, Scalar.RawSignificand: FixedWidthInteger {
   /// Returns a vector with random values from within the specified range in
@@ -655,6 +686,8 @@ where Scalar: BinaryFloatingPoint, Scalar.RawSignificand: FixedWidthInteger {
   }
 }
 
+#endif
+
 @frozen
 public struct SIMDMask<Storage>: SIMD
                   where Storage: SIMD,
@@ -695,6 +728,8 @@ public struct SIMDMask<Storage>: SIMD
   }
 }
 
+#if !$Embedded
+
 extension SIMDMask {
   /// Returns a vector mask with `true` or `false` randomly assigned in each
   /// lane, using the given generator as a source for randomness.
@@ -714,6 +749,8 @@ extension SIMDMask {
   }
 }
 
+#endif
+
 //  Implementations of integer operations. These should eventually all
 //  be replaced with @_semantics to lower directly to vector IR nodes.
 extension SIMD where Scalar: FixedWidthInteger {

stdlib/public/core/SIMDVectorTypes.swift.gyb

@@ -155,6 +155,8 @@ extension SIMD${n} where Scalar: FixedWidthInteger {
   }
 }
 
+#if !$Embedded
+
 extension SIMD${n}: CustomDebugStringConvertible {
   public var debugDescription: String {
     return "SIMD${n}<\(Scalar.self)>(${', '.join(map(lambda c:
@@ -163,6 +165,8 @@ extension SIMD${n}: CustomDebugStringConvertible {
   }
 }
 
+#endif
+
 extension SIMD${n} where Scalar: BinaryFloatingPoint {
   /// Creates a new vector from the given vector of integers.
   ///