pull swift/ and objc/ folder contents

2023-07-11 00:06:58 -07:00
parent 038da1b01a
commit bfd9340bfe
52 changed files with 3607 additions and 0 deletions
@@ -0,0 +1,62 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+import XCTest
+import Foundation
+@testable import OnnxRuntimeBindings
+
+final class SwiftOnnxRuntimeBindingsTests: XCTestCase {
+    let modelPath: String = Bundle.module.url(forResource: "single_add.basic", withExtension: "ort")!.path
+
+    func testGetVersionString() throws {
+        do {
+            let version = ORTVersion()
+            XCTAssertNotNil(version)
+        } catch let error {
+            XCTFail(error.localizedDescription)
+        }
+    }
+
+    func testCreateSession() throws {
+        do {
+            let env = try ORTEnv(loggingLevel: ORTLoggingLevel.verbose)
+            let options = try ORTSessionOptions()
+            try options.setLogSeverityLevel(ORTLoggingLevel.verbose)
+            try options.setIntraOpNumThreads(1)
+            // Create the ORTSession
+            _ = try ORTSession(env: env, modelPath: modelPath, sessionOptions: options)
+        } catch let error {
+            XCTFail(error.localizedDescription)
+        }
+    }
+
+    func testAppendCoreMLEP() throws {
+        do {
+            let env = try ORTEnv(loggingLevel: ORTLoggingLevel.verbose)
+            let sessionOptions: ORTSessionOptions = try ORTSessionOptions()
+            let coreMLOptions: ORTCoreMLExecutionProviderOptions = ORTCoreMLExecutionProviderOptions()
+            coreMLOptions.enableOnSubgraphs = true
+            try sessionOptions.appendCoreMLExecutionProvider(with: coreMLOptions)
+
+            XCTAssertTrue(ORTIsCoreMLExecutionProviderAvailable())
+            _ = try ORTSession(env: env, modelPath: modelPath, sessionOptions: sessionOptions)
+        } catch let error {
+            XCTFail(error.localizedDescription)
+        }
+    }
+
+    func testAppendXnnpackEP() throws {
+        do {
+            let env = try ORTEnv(loggingLevel: ORTLoggingLevel.verbose)
+            let sessionOptions: ORTSessionOptions = try ORTSessionOptions()
+            let XnnpackOptions: ORTXnnpackExecutionProviderOptions = ORTXnnpackExecutionProviderOptions()
+            XnnpackOptions.intra_op_num_threads = 2
+            try sessionOptions.appendXnnpackExecutionProvider(with: XnnpackOptions)
+
+            XCTAssertTrue(ORTIsCoreMLExecutionProviderAvailable())
+            _ = try ORTSession(env: env, modelPath: modelPath, sessionOptions: sessionOptions)
+        } catch let error {
+            XCTFail(error.localizedDescription)
+        }
+    }
+}