up

metascroy · metascroy · commit 73fa0833513d · 2025-04-29T12:23:43.000-07:00
diff --git a/backends/apple/coreml/runtime/delegate/coreml_backend_delegate.mm b/backends/apple/coreml/runtime/delegate/coreml_backend_delegate.mm
@@ -88,13 +88,12 @@
         ET_LOG(Error, "%s: DataType=%d is not supported", ETCoreMLStrings.delegateIdentifier.UTF8String, (int)tensor.scalar_type());
         return std::nullopt;
     }
-    
     std::vector<ssize_t> strides(tensor.strides().begin(), tensor.strides().end());
     std::vector<size_t> shape(tensor.sizes().begin(), tensor.sizes().end());
     
     // If tensor is rank 0, wrap in rank 1
     // See https://github.com/apple/coremltools/blob/8.2/coremltools/converters/mil/frontend/torch/exir_utils.py#L73
-    if (strides.size() == 0) {
+    if (shape.size() == 0) {
         shape.push_back(1);
         strides.push_back(1);
     }

Original file line number	Diff line number	Diff line change
`@@ -88,13 +88,12 @@`
`88`	`88`	`ET_LOG(Error, "%s: DataType=%d is not supported", ETCoreMLStrings.delegateIdentifier.UTF8String, (int)tensor.scalar_type());`
`89`	`89`	`return std::nullopt;`
`90`	`90`	`}`
`91`		`-`
`92`	`91`	`std::vector<ssize_t> strides(tensor.strides().begin(), tensor.strides().end());`
`93`	`92`	`std::vector<size_t> shape(tensor.sizes().begin(), tensor.sizes().end());`
`94`	`93`
`95`	`94`	`// If tensor is rank 0, wrap in rank 1`
`96`	`95`	`// See https://github.com/apple/coremltools/blob/8.2/coremltools/converters/mil/frontend/torch/exir_utils.py#L73`
`97`		`- if (strides.size() == 0) {`
	`96`	`+ if (shape.size() == 0) {`
`98`	`97`	`shape.push_back(1);`
`99`	`98`	`strides.push_back(1);`
`100`	`99`	`}`