whisper.swiftui : get device name / os from UIDevice

Author: jhen0409

examples/whisper.swiftui/whisper.cpp.swift/LibWhisper.swift

@@ -1,4 +1,5 @@
 import Foundation
+import UIKit
 import whisper
 
 enum WhisperError: Error {
@@ -64,67 +65,69 @@ actor WhisperContext {
     }
 
     func benchFull(modelName: String) async -> String {
-        let n_threads = Int32(min(4, cpuCount())) // Default in whisper.cpp
+        let nThreads = Int32(min(4, cpuCount())) // Default in whisper.cpp
 
-        let n_mels = whisper_model_n_mels(context)
-        if (whisper_set_mel(context, nil, 0, n_mels) != 0) {
+        let nMels = whisper_model_n_mels(context)
+        if (whisper_set_mel(context, nil, 0, nMels) != 0) {
             return "error: failed to set mel"
         }
 
         // heat encoder
-        if (whisper_encode(context, 0, n_threads) != 0) {
+        if (whisper_encode(context, 0, nThreads) != 0) {
             return "error: failed to encode"
         }
 
         var tokens = [whisper_token](repeating: 0, count: 512)
 
         // prompt heat
-        if (whisper_decode(context, &tokens, 256, 0, n_threads) != 0) {
+        if (whisper_decode(context, &tokens, 256, 0, nThreads) != 0) {
             return "error: failed to decode"
         }
 
         // text-generation heat
-        if (whisper_decode(context, &tokens, 1, 256, n_threads) != 0) {
+        if (whisper_decode(context, &tokens, 1, 256, nThreads) != 0) {
             return "error: failed to decode"
         }
 
         whisper_reset_timings(context)
 
         // actual run
-        if (whisper_encode(context, 0, n_threads) != 0) {
+        if (whisper_encode(context, 0, nThreads) != 0) {
             return "error: failed to encode"
         }
 
         // text-generation
         for i in 0..<256 {
-            if (whisper_decode(context, &tokens, 1, Int32(i), n_threads) != 0) {
+            if (whisper_decode(context, &tokens, 1, Int32(i), nThreads) != 0) {
                 return "error: failed to decode"
             }
         }
 
         // batched decoding
         for _ in 0..<64 {
-            if (whisper_decode(context, &tokens, 5, 0, n_threads) != 0) {
+            if (whisper_decode(context, &tokens, 5, 0, nThreads) != 0) {
                 return "error: failed to decode"
             }
         }
 
         // prompt processing
         for _ in 0..<16 {
-            if (whisper_decode(context, &tokens, 256, 0, n_threads) != 0) {
+            if (whisper_decode(context, &tokens, 256, 0, nThreads) != 0) {
                 return "error: failed to decode"
             }
         }
 
         whisper_print_timings(context)
 
-        let system_info = self.systemInfo()
+        let deviceModel = UIDevice.current.model
+        let systemName = UIDevice.current.systemName
+        let systemInfo = self.systemInfo()
         let timings: whisper_timings = whisper_get_timings(context).pointee
-        let encode_ms = String(format: "%.2f", timings.encode_ms)
-        let decode_ms = String(format: "%.2f", timings.decode_ms)
-        let batchd_ms = String(format: "%.2f", timings.batchd_ms)
-        let prompt_ms = String(format: "%.2f", timings.prompt_ms)
-        return "| <todo> | iOS | \(system_info) | \(modelName) | \(n_threads) | 1 | \(encode_ms) | \(decode_ms) | \(batchd_ms) | \(prompt_ms) | <todo> |"
+        let encodeMs = String(format: "%.2f", timings.encode_ms)
+        let decodeMs = String(format: "%.2f", timings.decode_ms)
+        let batchdMs = String(format: "%.2f", timings.batchd_ms)
+        let promptMs = String(format: "%.2f", timings.prompt_ms)
+        return "| \(deviceModel) | \(systemName) | \(systemInfo) | \(modelName) | \(nThreads) | 1 | \(encodeMs) | \(decodeMs) | \(batchdMs) | \(promptMs) | <todo> |"
     }
 
     static func createContext(path: String) throws -> WhisperContext {