Skip to content

Commit 0a5d91a

Browse files
authored
Update ETCoreMLModelManagerTests.mm
1 parent 1e574e8 commit 0a5d91a

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

backends/apple/coreml/runtime/test/ETCoreMLModelManagerTests.mm

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ - (void)testModelPrewarm {
9797
- (void)testAddModelExecution {
9898
NSURL *modelURL = [[self class] bundledResourceWithName:@"add_coreml_all" extension:@"bin"];
9999
XCTAssertNotNil(modelURL);
100-
100+
101101
NSError *localError = nil;
102102
NSData *data = [NSData dataWithContentsOfURL:modelURL];
103103
MLModelConfiguration *configuration = [[MLModelConfiguration alloc] init];
@@ -108,12 +108,12 @@ - (void)testAddModelExecution {
108108
int y = 50;
109109
// add_coreml_all does the following operation.
110110
int z = x + y;
111-
111+
112112
NSArray<MLMultiArray *> *inputs = [ETCoreMLTestUtils inputsForModel:model repeatedValues:@[@(x), @(y)] error:&localError];
113113
XCTAssertNotNil(inputs);
114114
MLMultiArray *output = [ETCoreMLTestUtils filledMultiArrayWithShape:inputs[0].shape dataType:inputs[0].dataType repeatedValue:@(0) error:&localError];
115115
NSArray<MLMultiArray *> *args = [inputs arrayByAddingObject:output];
116-
XCTAssertTrue([self.modelManager executeModelWithHandle:handle
116+
XCTAssertTrue([self.modelManager executeModelWithHandle:handle
117117
args:args
118118
loggingOptions:executorchcoreml::ModelLoggingOptions()
119119
eventLogger:nullptr
@@ -127,7 +127,7 @@ - (void)testAddModelExecution {
127127
- (void)testMulModelExecution {
128128
NSURL *modelURL = [[self class] bundledResourceWithName:@"mul_coreml_all" extension:@"bin"];
129129
XCTAssertNotNil(modelURL);
130-
130+
131131
NSError *localError = nil;
132132
NSData *data = [NSData dataWithContentsOfURL:modelURL];
133133
MLModelConfiguration *configuration = [[MLModelConfiguration alloc] init];
@@ -151,6 +151,7 @@ - (void)testMulModelExecution {
151151
}
152152
}
153153

154+
// See https://github.com/pytorch/executorch/pull/10465
154155
- (void)testAutoreleasepoolError {
155156
NSURL *modelURL = [self.class bundledResourceWithName:@"add_coreml_all" extension:@"bin"];
156157
NSError *localError = nil;

0 commit comments

Comments
 (0)