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