Fix Swift format

Author: stikves

swift/Sources/CoreAILanguageModels/Samplers/MPSGraphSamplers.swift

@@ -269,10 +269,12 @@ final class MPSGraphArgmaxSampler: @unchecked Sendable {
         self.bitmaskSize = (vocabSize + 31) / 32
 
         // Allocate bitmask buffer (shared memory -- CPU writes visible to GPU without DMA)
-        guard let bitmaskBuf = device.makeBuffer(
-            length: bitmaskSize * MemoryLayout<Int32>.size,
-            options: .storageModeShared
-        ) else {
+        guard
+            let bitmaskBuf = device.makeBuffer(
+                length: bitmaskSize * MemoryLayout<Int32>.size,
+                options: .storageModeShared
+            )
+        else {
             throw MPSGraphSamplerError.bufferAllocationFailed
         }
         self.bitmaskBuffer = bitmaskBuf
@@ -385,9 +387,10 @@ final class MPSGraphArgmaxSampler: @unchecked Sendable {
         outputOffset: Int,
         completion: @escaping (Int32) -> Void
     ) {
-        encode(to: queue, logitsBuffer: logitsBuffer, logitsOffset: logitsOffset,
-               outputBuffer: outputBuffer, outputOffset: outputOffset,
-               applyBitmask: false, completion: completion)
+        encode(
+            to: queue, logitsBuffer: logitsBuffer, logitsOffset: logitsOffset,
+            outputBuffer: outputBuffer, outputOffset: outputOffset,
+            applyBitmask: false, completion: completion)
     }
 
     /// Encode argmax sampling with optional bitmask constraint.
@@ -481,9 +484,10 @@ final class MPSGraphArgmaxSampler: @unchecked Sendable {
         outputOffset: Int,
         completion: @escaping (Int32) -> Void
     ) {
-        encodeWithSlice(to: queue, logitsBuffer: logitsBuffer, queryLength: queryLength,
-                        outputBuffer: outputBuffer, outputOffset: outputOffset,
-                        applyBitmask: false, completion: completion)
+        encodeWithSlice(
+            to: queue, logitsBuffer: logitsBuffer, queryLength: queryLength,
+            outputBuffer: outputBuffer, outputOffset: outputOffset,
+            applyBitmask: false, completion: completion)
     }
 
     /// Encode argmax with slice support and optional bitmask.
@@ -908,9 +912,10 @@ final class MPSGraphCompositeSampler: @unchecked Sendable {
         outputOffset: Int,
         completion: @escaping (Int32) -> Void
     ) {
-        encode(to: queue, logitsBuffer: logitsBuffer, logitsOffset: logitsOffset,
-               outputBuffer: outputBuffer, outputOffset: outputOffset,
-               applyBitmask: false, completion: completion)
+        encode(
+            to: queue, logitsBuffer: logitsBuffer, logitsOffset: logitsOffset,
+            outputBuffer: outputBuffer, outputOffset: outputOffset,
+            applyBitmask: false, completion: completion)
     }
 
     /// Encode composite sampling with optional bitmask constraint.
@@ -1003,9 +1008,10 @@ final class MPSGraphCompositeSampler: @unchecked Sendable {
         outputOffset: Int,
         completion: @escaping (Int32) -> Void
     ) {
-        encodeWithSlice(to: queue, logitsBuffer: logitsBuffer, queryLength: queryLength,
-                        outputBuffer: outputBuffer, outputOffset: outputOffset,
-                        applyBitmask: false, completion: completion)
+        encodeWithSlice(
+            to: queue, logitsBuffer: logitsBuffer, queryLength: queryLength,
+            outputBuffer: outputBuffer, outputOffset: outputOffset,
+            applyBitmask: false, completion: completion)
     }
 
     /// Encode composite sampling with slice and optional bitmask.

swift/Tests/GuidedGenerationTests/ConstrainedGenerationSessionTests.swift

@@ -103,8 +103,9 @@ struct ConstrainedGenerationSessionTests {
         // They should be identical
         #expect(arrayBitmask!.count == pointerBitmask.count)
         for i in 0..<arrayBitmask!.count {
-            #expect(arrayBitmask![i] == pointerBitmask[i],
-                    "Mismatch at word \(i): array=\(arrayBitmask![i]) pointer=\(pointerBitmask[i])")
+            #expect(
+                arrayBitmask![i] == pointerBitmask[i],
+                "Mismatch at word \(i): array=\(arrayBitmask![i]) pointer=\(pointerBitmask[i])")
         }
     }
 

swift/Tests/LanguageModelsTests/MPSGraphSamplerTests.swift

@@ -907,8 +907,9 @@ struct MPSGraphConstrainedArgmaxTests {
         }
         let unconstrainedResult = outputBuffer.contents().assumingMemoryBound(to: Int32.self).pointee
 
-        #expect(constrainedResult == unconstrainedResult,
-                "All-ones bitmask should match unconstrained: \(constrainedResult) vs \(unconstrainedResult)")
+        #expect(
+            constrainedResult == unconstrainedResult,
+            "All-ones bitmask should match unconstrained: \(constrainedResult) vs \(unconstrainedResult)")
     }
 
     @Test("Bitmask allows only 3 tokens -- argmax picks best among them")
@@ -956,7 +957,8 @@ struct MPSGraphConstrainedCompositeTests {
     @Test("Bitmask blocks dominant token -- composite never samples it")
     func bitmaskBlocksDominantComposite() async throws {
         let device = try #require(Self.device)
-        let sampler = try MPSGraphCompositeSampler(device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
+        let sampler = try MPSGraphCompositeSampler(
+            device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
 
         let logitsBuffer = try #require(device.makeBuffer(length: Self.vocabSize * 2, options: .storageModeShared))
         let outputBuffer = try #require(device.makeBuffer(length: 4, options: .storageModeShared))
@@ -995,7 +997,8 @@ struct MPSGraphConstrainedCompositeTests {
     @Test("Only 3 tokens allowed -- composite samples exclusively from them")
     func compositeOnlyAllowedTokens() async throws {
         let device = try #require(Self.device)
-        let sampler = try MPSGraphCompositeSampler(device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
+        let sampler = try MPSGraphCompositeSampler(
+            device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
 
         let logitsBuffer = try #require(device.makeBuffer(length: Self.vocabSize * 2, options: .storageModeShared))
         let outputBuffer = try #require(device.makeBuffer(length: 4, options: .storageModeShared))
@@ -1036,7 +1039,8 @@ struct MPSGraphConstrainedCompositeTests {
     @Test("Constrained composite latency under 25ms for 32K vocab")
     func constrainedCompositePerformance() async throws {
         let device = try #require(Self.device)
-        let sampler = try MPSGraphCompositeSampler(device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
+        let sampler = try MPSGraphCompositeSampler(
+            device: device, vocabSize: Self.vocabSize, k: Self.k, temperature: 1.0)
 
         let logitsBuffer = try #require(device.makeBuffer(length: Self.vocabSize * 2, options: .storageModeShared))
         let outputBuffer = try #require(device.makeBuffer(length: 4, options: .storageModeShared))