Fix expand pattern when strides of src/dst are not matching (#203)

kulinseth · Dec 19, 2022 · 506fa5b · 506fa5b
1 parent a023830
commit 506fa5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aten/src/ATen/native/mps/operations/View.mm b/aten/src/ATen/native/mps/operations/View.mm
@@ -145,7 +145,7 @@
     NSUInteger targetDimLength =  currSrcDimLength;
     if (currDimLength != targetDimLength)
       targetDimLength = 1;
-    if (currDimLength != targetDimLength && currStride != currSrcStride)
+    if (currDimLength != targetDimLength || currStride != currSrcStride)
       isValidExpand = NO;
     if (currSrcDim >= 0 && currSrcDimLength == targetDimLength) {
       currSrcStride *= currSrcDimLength;