Update

triton-lang · Aug 9, 2024 · 593e5fa · 593e5fa
1 parent 31ad7d6
commit 593e5fa
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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;
  }