v2: add embedding tests, multilingual embedding support

2026-07-06 08:01:52 +08:00
parent ba4c41c29f
commit 85dd87e28a
13 changed files with 484 additions and 15 deletions
@@ -0,0 +1,33 @@
+import Foundation
+import XCTest
+
+@testable import MarkBase
+
+final class Timing26BTest: XCTestCase {
+    let modelDir = "/Users/accusys/MarkBaseEngine/models/gemma-4-26b-standard"
+    var engine: MarkBaseEngine!
+    var model: E4BModel!
+    let maxCtx = 128
+
+    override func setUp() {
+        super.setUp()
+        guard FileManager.default.fileExists(atPath: modelDir + "/model.safetensors") else {
+            return
+        }
+        let t0 = Date()
+        engine = try? MarkBaseEngine(autoCompile: true)
+        model = try? E4BModel(modelDir: modelDir, engine: engine, maxContextLength: maxCtx)
+        let loadTime = Date().timeIntervalSince(t0)
+        print("Total init: \(String(format: "%.1f", loadTime))s")
+    }
+
+    func testForwardTiming() throws {
+        try XCTSkipIf(model == nil, "26B model not found")
+        for i in 0..<5 {
+            let t = Date()
+            _ = try model.forward(tokenId: 100 + i, position: i)
+            let ft = Date().timeIntervalSince(t)
+            print("Forward \(i+1): \(String(format: "%.3f", ft))s")
+        }
+    }
+}