diff --git a/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp b/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp
index 7f3bed5f2efb..e4d7c8e363f8 100644
--- a/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp
+++ b/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp
@@ -162,7 +162,7 @@ LinearLayout ensureLayoutNotLargerThan(
       continue;
     }
     assert(actualSize % desiredSize == 0);
-    // <inDimName, baseIdx, baseValue>
+    // <inDimName, baseIdx, outValue>
     std::vector<std::tuple<StringAttr, int, int>> sortedBases;
     for (auto [inDimName, basis] : bases) {
       for (size_t baseIdx = 0; baseIdx < basis.size(); baseIdx++) {
@@ -177,9 +177,8 @@ LinearLayout ensureLayoutNotLargerThan(
     llvm::sort(sortedBases,
                [](auto a, auto b) { return std::get<2>(a) > std::get<2>(b); });
     for (auto [inDimName, basisIdx, outValue] : sortedBases) {
-      if (actualSize <= desiredSize) {
+      if (actualSize <= desiredSize)
         break;
-      }
       bases[inDimName][basisIdx][outDim.index()] = 0;
       actualSize >>= 1;
     }