Skip to content

Commit 97d62c0

Browse files
committed
fix broken test
1 parent 65171cc commit 97d62c0

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

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

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#import <XCTest/XCTest.h>
1616
#import <executorch/runtime/platform/runtime.h>
1717
#import <model_logging_options.h>
18+
#import <multiarray.h>
1819

1920
@interface ETCoreMLModelManagerTests : XCTestCase
2021

@@ -169,10 +170,10 @@ - (void)testAutoreleasepoolError {
169170
[ETCoreMLTestUtils inputsForModel:model repeatedValues:@[@(2), @(3)] error:&localError];
170171
XCTAssert(inputArrays);
171172

172-
std::vector<MultiArray> multiArrays;
173+
std::vector<executorchcoreml::MultiArray> multiArrays;
173174
multiArrays.reserve(inputArrays.count + model.orderedOutputNames.count);
174175
for (MLMultiArray *array in inputArrays) {
175-
auto dataTypeOpt = to_multiarray_data_type(array.dataType);
176+
auto dataTypeOpt = executorchcoreml::to_multiarray_data_type(array.dataType);
176177
XCTAssert(dataTypeOpt.has_value());
177178
auto dataType = dataTypeOpt.value();
178179

@@ -189,7 +190,7 @@ - (void)testAutoreleasepoolError {
189190
}
190191

191192
multiArrays.emplace_back(array.dataPointer,
192-
MultiArray::MemoryLayout(dataType, dims, strides));
193+
executorchcoreml::MultiArray::MemoryLayout(dataType, dims, strides));
193194
}
194195

195196
auto inputLayout = multiArrays[0].layout();
@@ -202,15 +203,15 @@ - (void)testAutoreleasepoolError {
202203
auto originalLayout = multiArrays[0].layout();
203204
auto corruptedDims = originalLayout.shape();
204205
corruptedDims[0] += 1;
205-
multiArrays[0] = MultiArray(multiArrays[0].data(),
206-
MultiArray::MemoryLayout(originalLayout.dataType(),
206+
multiArrays[0] = executorchcoreml::MultiArray(multiArrays[0].data(),
207+
executorchcoreml::MultiArray::MemoryLayout(originalLayout.dataType(),
207208
corruptedDims,
208209
originalLayout.strides()));
209210
}
210211

211212
BOOL success = [self.modelManager executeModelWithHandle:modelHandle
212213
argsVec:multiArrays
213-
loggingOptions:ModelLoggingOptions()
214+
loggingOptions:executorchcoreml::ModelLoggingOptions()
214215
eventLogger:nullptr
215216
error:&localError];
216217
XCTAssertFalse(success);

0 commit comments

Comments
 (0)