fix: Change types in ETModel.mm

software-mansion · Dec 5, 2024 · 0f718b8 · 0f718b8
1 parent 9c93888
commit 0f718b8
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/third-party/ios/ExecutorchLib/ExecutorchLib/Exported/ETModel.mm b/third-party/ios/ExecutorchLib/ExecutorchLib/Exported/ETModel.mm
@@ -44,27 +44,27 @@ - (NSArray *)forward:(NSArray *)input
     }
     case 1: {
       // Int32Array
-      std::vector<DataPtrWithNumel<int8_t>> output =
-          runForwardFromNSArray<int8_t>(input, shapes, _model);
-      return arrayToNSArray<int8_t>(output);
+      std::vector<DataPtrWithNumel<int32_t>> output =
+          runForwardFromNSArray<int32_t>(input, shapes, _model);
+      return arrayToNSArray<int32_t>(output);
     }
     case 2: {
       // BigInt64Array
-      std::vector<DataPtrWithNumel<int8_t>> output =
-          runForwardFromNSArray<int8_t>(input, shapes, _model);
-      return arrayToNSArray<int8_t>(output);
+      std::vector<DataPtrWithNumel<int64_t>> output =
+          runForwardFromNSArray<int64_t>(input, shapes, _model);
+      return arrayToNSArray<int64_t>(output);
     }
     case 3: {
       // Float32Array
-      std::vector<DataPtrWithNumel<int8_t>> output =
-          runForwardFromNSArray<int8_t>(input, shapes, _model);
-      return arrayToNSArray<int8_t>(output);
+      std::vector<DataPtrWithNumel<float>> output =
+          runForwardFromNSArray<float>(input, shapes, _model);
+      return arrayToNSArray<float>(output);
     }
     case 4: {
       // Float64Array
-      std::vector<DataPtrWithNumel<int8_t>> output =
-          runForwardFromNSArray<int8_t>(input, shapes, _model);
-      return arrayToNSArray<int8_t>(output);
+      std::vector<DataPtrWithNumel<double>> output =
+          runForwardFromNSArray<double>(input, shapes, _model);
+      return arrayToNSArray<double>(output);
     }
     }
   } @catch (NSException *exception) {