Improve readability for isInVnniLayout (NFC) (#808)

plaidml · Nov 27, 2023 · 45d00a1 · 45d00a1
1 parent 95e4a7d
commit 45d00a1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lib/TPP/Transforms/Utils/VNNIUtils.cpp b/lib/TPP/Transforms/Utils/VNNIUtils.cpp
@@ -27,8 +27,7 @@ std::optional<int64_t> getVnniBlockingFactor(Type type) {
 bool isInVnniLayout(MemRefType memref) {
   if (memref.getRank() < 3 || !memref.getElementType().isBF16())
     return false;
-  return memref.getShape()[memref.getRank() - 1] ==
-         vnni::utils::getVnniBlockingFactor(memref);
+  return memref.getShape().back() == vnni::utils::getVnniBlockingFactor(memref);
 }
 
 } // namespace utils