94 lines
2.9 KiB
Swift
94 lines
2.9 KiB
Swift
// Copyright 2019 Google Inc. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at:
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
import XCTest
|
|
|
|
@testable import TensorFlowLite
|
|
|
|
class MetalDelegateTests: XCTestCase {
|
|
|
|
func testInitDefaultGPUDelegateOptions() {
|
|
let delegate = MetalDelegate()
|
|
XCTAssertFalse(delegate.options.allowsPrecisionLoss)
|
|
XCTAssertEqual(delegate.options.waitType, .passive)
|
|
}
|
|
|
|
func testInitWithCustomGPUDelegateOptions() {
|
|
var options = MetalDelegate.Options()
|
|
options.allowsPrecisionLoss = true
|
|
options.waitType = .active
|
|
let delegate = MetalDelegate(options: options)
|
|
XCTAssertTrue(delegate.options.allowsPrecisionLoss)
|
|
XCTAssertEqual(delegate.options.waitType, .active)
|
|
}
|
|
|
|
func testInitInterpreterWithDelegate() throws {
|
|
let metalDelegate = MetalDelegate()
|
|
let interpreter = try Interpreter(modelPath: AddQuantizedModel.path, delegates: [metalDelegate])
|
|
XCTAssertEqual(interpreter.delegates?.count, 1)
|
|
XCTAssertNil(interpreter.options)
|
|
}
|
|
|
|
func testInitInterpreterWithOptionsAndDelegate() throws {
|
|
var options = Interpreter.Options()
|
|
options.threadCount = 1
|
|
let metalDelegate = MetalDelegate()
|
|
let interpreter = try Interpreter(
|
|
modelPath: AddQuantizedModel.path,
|
|
options: options,
|
|
delegates: [metalDelegate]
|
|
)
|
|
XCTAssertNotNil(interpreter.options)
|
|
XCTAssertEqual(interpreter.delegates?.count, 1)
|
|
}
|
|
}
|
|
|
|
class MetalDelegateOptionsTests: XCTestCase {
|
|
|
|
func testInitWithDefaultValues() {
|
|
let options = MetalDelegate.Options()
|
|
XCTAssertFalse(options.allowsPrecisionLoss)
|
|
XCTAssertEqual(options.waitType, .passive)
|
|
}
|
|
|
|
func testInitWithCustomValues() {
|
|
var options = MetalDelegate.Options()
|
|
options.allowsPrecisionLoss = true
|
|
options.waitType = .active
|
|
XCTAssertTrue(options.allowsPrecisionLoss)
|
|
XCTAssertEqual(options.waitType, .active)
|
|
}
|
|
|
|
func testEquatable() {
|
|
var options1 = MetalDelegate.Options()
|
|
var options2 = MetalDelegate.Options()
|
|
XCTAssertEqual(options1, options2)
|
|
|
|
options1.allowsPrecisionLoss = true
|
|
options2.allowsPrecisionLoss = true
|
|
XCTAssertEqual(options1, options2)
|
|
|
|
options1.waitType = .none
|
|
options2.waitType = .none
|
|
XCTAssertEqual(options1, options2)
|
|
|
|
options2.allowsPrecisionLoss = false
|
|
XCTAssertNotEqual(options1, options2)
|
|
options1.allowsPrecisionLoss = false
|
|
|
|
options1.waitType = .aggressive
|
|
XCTAssertNotEqual(options1, options2)
|
|
}
|
|
}
|