Tests for uint vectors in simd.

Author: stephentyrone

test/1_stdlib/simd.swift.gyb

@@ -19,9 +19,9 @@ import SwiftPrivate
 import ObjectiveC
 #endif
 
-% scalar_types = ['Float', 'Double', 'Int32']
+% scalar_types = ['Float', 'Double', 'Int32', 'UInt32']
 % float_types = ['Float', 'Double']
-% ctype = { 'Float':'float', 'Double':'double', 'Int32':'int' }
+% ctype = { 'Float':'float', 'Double':'double', 'Int32':'int', 'UInt32':'uint'}
 % component = ['.x','.y','.z','.w']
 
 % for type in scalar_types:
@@ -146,12 +146,12 @@ simdTestSuite.test("vector elements") {
 
   //  Check getting and setting .xyzw
 % for i in range(size):
-  ${vec + component[i]} = ${-i}
-  expectEqual(${vec + component[i]}, ${-i})
+  ${vec + component[i]} = ${2*i}
+  expectEqual(${vec + component[i]}, ${2*i})
 % end
 
   //  Check getting and setting [i]
-  for i in 0..<${size} { ${vec}[i] = -${vec}[i] }
+  for i in 0..<${size} { ${vec}[i] = ${vec}[i]/2 }
   expectEqual(${vec}, ${range(size)}, sameValue: same)
 
 %   end # for size in [2,3,4]

test/ClangModules/simd.swift

@@ -6,10 +6,12 @@ import simd
 let f4: float4 = makes_float4()
 let i3: int3 = makes_int3()
 let d2: double2 = makes_double2()
+let u4: uint4 = makes_uint4()
 
 takes_float4(f4)
 takes_int3(i3)
 takes_double2(d2)
+takes_uint4(u4)
 
 // byte17 isn't available since there isn't a type in the simd module to map it to.
 

test/Inputs/clang-importer-sdk/swift-modules-without-ns/simd.swift

@@ -2,3 +2,4 @@
 @_alignment(16) public struct float4 { public var x, y, z, w: Float }
 @_alignment(16) public struct double2 { public var x, y: Double }
 @_alignment(16) public struct int3 { public var x, y, z: Int32 }
+@_alignment(16) public struct uint4 { public var x, y, z, w: UInt32 }

test/Inputs/clang-importer-sdk/swift-modules/simd.swift

@@ -2,3 +2,4 @@
 @_alignment(16) public struct float4 { public var x, y, z, w: Float }
 @_alignment(16) public struct double2 { public var x, y: Double }
 @_alignment(16) public struct int3 { public var x, y, z: Int32 }
+@_alignment(16) public struct uint4 { public var x, y, z, w: UInt32 }

test/Inputs/clang-importer-sdk/usr/include/simd.h

@@ -2,15 +2,18 @@ typedef int int3 __attribute__((__ext_vector_type__(3)));
 typedef float float3 __attribute__((__ext_vector_type__(3)));
 typedef float float4 __attribute__((__ext_vector_type__(4)));
 typedef double double2 __attribute__((__ext_vector_type__(2)));
+typedef unsigned int uint4 __attribute__((__ext_vector_type__(4)));
 typedef unsigned char byte17 __attribute__((__ext_vector_type__(17)));
 
 int3    makes_int3(void);
+uint4   makes_uint4(void);
 float3  makes_float3(void);
 float4  makes_float4(void);
 double2 makes_double2(void);
 byte17  makes_byte17(void);
 
 void takes_int3(int3 x);
+void takes_uint4(uint4 x);
 void takes_float3(float3 x);
 void takes_float4(float4 x);
 void takes_double2(double2 x);

test/PrintAsObjC/simd.swift

@@ -22,6 +22,7 @@ import simd
 // CHECK-LABEL: typedef float swift_float4 __attribute__((__ext_vector_type__(4)));
 // CHECK-LABEL: typedef double swift_double2 __attribute__((__ext_vector_type__(2)));
 // CHECK-LABEL: typedef int swift_int3 __attribute__((__ext_vector_type__(3)));
+// CHECK-LABEL: typedef unsigned int swift_uint4 __attribute__((__ext_vector_type__(4)));
 
 // -- The C simd module is useless to Swift.
 // CHECK-NOT: @import simd;
@@ -34,5 +35,7 @@ import simd
   @objc func doStuffWithDouble2(x: double2) -> double2 { return x }
   // CHECK-LABEL: - (swift_int3)doStuffWithInt3:(swift_int3)x;
   @objc func doStuffWithInt3(x: int3) -> int3 { return x }
+  // CHECK-LABEL: - (swift_uint4)doStuffWithUInt4:(swift_uint4)x;
+  @objc func doStuffWithUInt4(x: uint4) -> uint4 { return x }
 }
 

test/attr/attr_objc_simd.swift

@@ -9,5 +9,6 @@ import simd
   @objc func doStuffWithFloat4(x: float4) -> float4 { return x }
   @objc func doStuffWithDouble2(x: double2) -> double2 { return x }
   @objc func doStuffWithInt3(x: int3) -> int3 { return x }
+  @objc func doStuffWithUInt4(x: uint4) -> uint4 { return x }
 }