[5.0] SIMD into stdlib (#20896)

* SIMD into stdlib

Implements SE-0229.

Also updates simd module types in the Apple SDKs to use the new types, and updates a couple tests to work with the new types and protocols.

(cherry picked from commit fb8b9e1)

* Move most of the simd operators into an optional module (#20914)

* Move most of the simd operators into an optional module

Adding simd to the stdlib caused some typechecker regressions. We can resolve them in the near-term by making the types universally available, but moving the arithmetic operators into a separate module that must be explicitly imported.

* Move two fuzzing tests back to fixed.

* Add SIMDOperators as a dependency for MediaPlayer.

* Move the .-prefixed operator declarations back into the stdlib.

(cherry picked from commit 28962b5)

Author: stephentyrone

include/swift/Runtime/BuiltinTypes.def

@@ -74,32 +74,43 @@ BUILTIN_VECTOR_TYPE(Bi16_, Int16, 4)
 BUILTIN_VECTOR_TYPE(Bi16_, Int16, 8)
 BUILTIN_VECTOR_TYPE(Bi16_, Int16, 16)
 BUILTIN_VECTOR_TYPE(Bi16_, Int16, 32)
+BUILTIN_VECTOR_TYPE(Bi16_, Int16, 64)
 
 // Int32 vector types
 BUILTIN_VECTOR_TYPE(Bi32_, Int32, 2)
 BUILTIN_VECTOR_TYPE(Bi32_, Int32, 3)
 BUILTIN_VECTOR_TYPE(Bi32_, Int32, 4)
 BUILTIN_VECTOR_TYPE(Bi32_, Int32, 8)
 BUILTIN_VECTOR_TYPE(Bi32_, Int32, 16)
+BUILTIN_VECTOR_TYPE(Bi32_, Int32, 32)
+BUILTIN_VECTOR_TYPE(Bi32_, Int32, 64)
 
 // Int64 vector types
 BUILTIN_VECTOR_TYPE(Bi64_, Int64, 2)
 BUILTIN_VECTOR_TYPE(Bi64_, Int64, 3)
 BUILTIN_VECTOR_TYPE(Bi64_, Int64, 4)
 BUILTIN_VECTOR_TYPE(Bi64_, Int64, 8)
+BUILTIN_VECTOR_TYPE(Bi64_, Int64, 16)
+BUILTIN_VECTOR_TYPE(Bi64_, Int64, 32)
+BUILTIN_VECTOR_TYPE(Bi64_, Int64, 64)
 
 // Float32 vector types
 BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 2)
 BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 3)
 BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 4)
 BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 8)
 BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 16)
+BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 32)
+BUILTIN_VECTOR_TYPE(Bf32_, FPIEEE32, 64)
 
 // Float64 vector types
 BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 2)
 BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 3)
 BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 4)
 BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 8)
+BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 16)
+BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 32)
+BUILTIN_VECTOR_TYPE(Bf64_, FPIEEE64, 64)
 
 #undef BUILTIN_VECTOR_TYPE
 #undef VECTOR_BUILTIN_SYMBOL_NAME

lib/IRGen/GenClangType.cpp

@@ -453,6 +453,7 @@ GenClangType::visitBoundGenericType(CanBoundGenericType type) {
     AutoreleasingUnsafeMutablePointer,
     Unmanaged,
     CFunctionPointer,
+    SIMD,
   } kind = llvm::StringSwitch<StructKind>(swiftStructDecl->getName().str())
     .Case("UnsafeMutablePointer", StructKind::UnsafeMutablePointer)
     .Case("UnsafePointer", StructKind::UnsafePointer)
@@ -461,6 +462,7 @@ GenClangType::visitBoundGenericType(CanBoundGenericType type) {
         StructKind::AutoreleasingUnsafeMutablePointer)
     .Case("Unmanaged", StructKind::Unmanaged)
     .Case("CFunctionPointer", StructKind::CFunctionPointer)
+    .StartsWith("SIMD", StructKind::SIMD)
     .Default(StructKind::Invalid);
 
   auto args = type.getGenericArgs();
@@ -472,7 +474,7 @@ GenClangType::visitBoundGenericType(CanBoundGenericType type) {
   case StructKind::Invalid:
     llvm_unreachable("Unexpected non-pointer generic struct type in imported"
                      " Clang module!");
-      
+    
   case StructKind::UnsafeMutablePointer:
   case StructKind::Unmanaged:
   case StructKind::AutoreleasingUnsafeMutablePointer: {
@@ -499,6 +501,19 @@ GenClangType::visitBoundGenericType(CanBoundGenericType type) {
     auto fnPtrTy = clangCtx.getPointerType(functionTy);
     return getCanonicalType(fnPtrTy);
   }
+    
+  case StructKind::SIMD: {
+    clang::QualType scalarTy = Converter.convert(IGM, loweredArgTy);
+    auto numEltsString = swiftStructDecl->getName().str();
+    numEltsString.consume_front("SIMD");
+    unsigned numElts;
+    bool failedParse = numEltsString.getAsInteger<unsigned>(10, numElts);
+    assert(!failedParse && "SIMD type name didn't end in count?");
+    (void) failedParse;
+    auto vectorTy = getClangASTContext().getVectorType(scalarTy, numElts,
+      clang::VectorType::VectorKind::GenericVector);
+    return getCanonicalType(vectorTy);
+  }
   }
 
   llvm_unreachable("Not a valid StructKind.");

stdlib/public/CMakeLists.txt

@@ -55,6 +55,7 @@ if(SWIFT_BUILD_STDLIB)
   add_subdirectory(runtime)
   add_subdirectory(stubs)
   add_subdirectory(core)
+  add_subdirectory(SIMDOperators)
   add_subdirectory(SwiftOnoneSupport)
 endif()
 

stdlib/public/SDK/MediaPlayer/CMakeLists.txt

@@ -8,7 +8,7 @@ add_swift_target_library(swiftMediaPlayer ${SWIFT_SDK_OVERLAY_LIBRARY_BUILD_TYPE
   LINK_FLAGS "${SWIFT_RUNTIME_SWIFT_LINK_FLAGS}"
   TARGET_SDKS IOS IOS_SIMULATOR
 
-  SWIFT_MODULE_DEPENDS_IOS Darwin AVFoundation CoreAudio CoreFoundation CoreGraphics CoreImage CoreMedia Dispatch Foundation Metal ObjectiveC QuartzCore simd UIKit os CoreData # auto-updated
+  SWIFT_MODULE_DEPENDS_IOS Darwin AVFoundation CoreAudio CoreFoundation CoreGraphics CoreImage CoreMedia Dispatch Foundation Metal ObjectiveC QuartzCore simd SIMDOperators UIKit os CoreData # auto-updated
   FRAMEWORK_DEPENDS_WEAK MediaPlayer
 
   DEPLOYMENT_VERSION_IOS ${SWIFTLIB_DEPLOYMENT_VERSION_MEDIAPLAYER_IOS}

stdlib/public/SDK/SceneKit/SceneKit.swift.gyb

@@ -51,13 +51,13 @@ extension SCNVector3 {
   }
 }
 
-extension float3 {
+extension SIMD3 where Scalar == Float {
   public init(_ v: SCNVector3) {
     self.init(Float(v.x), Float(v.y), Float(v.z))
   }
 }
 
-extension double3 {
+extension SIMD3 where Scalar == Double {
   public init(_ v: SCNVector3) {
     self.init(Double(v.x), Double(v.y), Double(v.z))
   }
@@ -86,13 +86,13 @@ extension SCNVector4 {
   }
 }
 
-extension float4 {
+extension SIMD4 where Scalar == Float {
   public init(_ v: SCNVector4) {
     self.init(Float(v.x), Float(v.y), Float(v.z), Float(v.w))
   }
 }
 
-extension double4 {
+extension SIMD4 where Scalar == Double {
   public init(_ v: SCNVector4) {
     self.init(Double(v.x), Double(v.y), Double(v.z), Double(v.w))
   }

stdlib/public/SDK/simd/Quaternion.swift.gyb

@@ -14,6 +14,7 @@
 
 import Swift
 import Darwin
+import SIMDOperators
 @_exported import simd
 
 %for scalar in ['Float','Double']: