[Fix] Fix logic error in dimension check (redundant check on mat2) (#171791)

lingebeng · pytorchmergebot · commit 7e5e018ef056 · 2026-01-06T18:17:16.000Z
Fix logic error in dimension check (redundant check on mat2) Pull Request resolved: #171791 Approved by: https://github.com/malfet
diff --git a/aten/src/ATen/native/mkldnn/Matmul.cpp b/aten/src/ATen/native/mkldnn/Matmul.cpp
@@ -423,7 +423,7 @@ static inline bool checksize(const Tensor& mat1, const Tensor& mat2){
   } else if (mat1.dim() == 2 && mat2.dim() == 1) {
     // aten::mv
     return mat1.size(0) * mat1.size(1) > mkldnn_gemm_min_size;
-  } else if (mat2.dim() == 2 && mat2.dim() == 2) {
+  } else if (mat1.dim() == 2 && mat2.dim() == 2) {
     // aten::addmm
     return mat1.size(0) * mat1.size(1) * mat2.size(1) > mkldnn_gemm_min_size;
   } else {
