format changes

Signed-off-by: bogdan-petkovic <bogdan.petkovic@htecgroup.com>

Author: bogdan-petkovic

mlir/lib/Dialect/Rock/IR/AmdArchDb.cpp

@@ -398,7 +398,8 @@ AmdArchInfo mlir::rock::lookupArchInfo(StringRef arch) {
   llvm_unreachable(msg.c_str());
 }
 
-FailureOr<int64_t> mlir::rock::lookupDeviceGlobalMemorySizeBytes(StringRef arch) {
+FailureOr<int64_t>
+mlir::rock::lookupDeviceGlobalMemorySizeBytes(StringRef arch) {
 #ifdef _WIN32
   (void)arch;
   return failure();
@@ -422,8 +423,8 @@ FailureOr<int64_t> mlir::rock::lookupDeviceGlobalMemorySizeBytes(StringRef arch)
     return getMemoryForDevice(deviceId);
 
   int deviceCount = 0;
-  if (auto err = hipGetDeviceCount(&deviceCount); err != hipSuccess ||
-                                                   deviceCount <= 0)
+  if (auto err = hipGetDeviceCount(&deviceCount);
+      err != hipSuccess || deviceCount <= 0)
     return failure();
 
   int64_t minMatchingBytes = std::numeric_limits<int64_t>::max();

mlir/lib/Dialect/Rock/IR/RockDialect.cpp

@@ -128,8 +128,7 @@ static FailureOr<int64_t> getTypeSizeInBytes(Type type) {
 }
 
 static FailureOr<int64_t> getSplitKVExtraStorageLimitOverrideBytes() {
-  const char *envVar =
-      std::getenv("ROCMLIR_ATTENTION_SPLITKV_MAX_EXTRA_BYTES");
+  const char *envVar = std::getenv("ROCMLIR_ATTENTION_SPLITKV_MAX_EXTRA_BYTES");
   if (!envVar || *envVar == '\0')
     return failure();
 
@@ -155,7 +154,8 @@ static int64_t getSplitKVExtraStorageLimitBytes(AttentionOp op) {
     return maybeOverride.value();
 
   StringAttr arch = rock::getArchValue(op.getOperation());
-  auto maybeDeviceBytes = rock::lookupDeviceGlobalMemorySizeBytes(arch.getValue());
+  auto maybeDeviceBytes =
+      rock::lookupDeviceGlobalMemorySizeBytes(arch.getValue());
   if (failed(maybeDeviceBytes) || maybeDeviceBytes.value() <= 0)
     return defaultLimitBytes;
 
@@ -189,9 +189,9 @@ static LogicalResult verifySplitKVExtraStorage(AttentionOp op,
     return success();
 
   auto maybeBaseElems = checkedMul(batchHeads, seqLenQ);
-  auto maybeBaseOutElems =
-      succeeded(maybeBaseElems) ? checkedMul(*maybeBaseElems, headDimV)
-                                : FailureOr<int64_t>(failure());
+  auto maybeBaseOutElems = succeeded(maybeBaseElems)
+                               ? checkedMul(*maybeBaseElems, headDimV)
+                               : FailureOr<int64_t>(failure());
   if (failed(maybeBaseElems) || failed(maybeBaseOutElems))
     return op.emitError("splitKV storage estimate overflowed");
 
@@ -203,16 +203,17 @@ static LogicalResult verifySplitKVExtraStorage(AttentionOp op,
   if (succeeded(maybeExtraOutBytes))
     maybeExtraOutBytes = checkedMul(*maybeExtraOutBytes, *maybeOutElemBytes);
 
-  auto maybeExtraLseBytes =
-      succeeded(maybeBaseElems) ? checkedMul(*maybeBaseElems, extraSplitFactor)
+  auto maybeExtraLseBytes = succeeded(maybeBaseElems)
+                                ? checkedMul(*maybeBaseElems, extraSplitFactor)
                                 : FailureOr<int64_t>(failure());
   if (succeeded(maybeExtraLseBytes))
     maybeExtraLseBytes = checkedMul(*maybeExtraLseBytes, *maybeLseElemBytes);
 
   if (failed(maybeExtraOutBytes) || failed(maybeExtraLseBytes))
     return op.emitError("splitKV storage estimate overflowed");
 
-  auto maybeTotalExtraBytes = checkedAdd(*maybeExtraOutBytes, *maybeExtraLseBytes);
+  auto maybeTotalExtraBytes =
+      checkedAdd(*maybeExtraOutBytes, *maybeExtraLseBytes);
   if (failed(maybeTotalExtraBytes))
     return op.emitError("splitKV storage estimate overflowed");
 
@@ -3335,8 +3336,8 @@ static LogicalResult verifyGemmPlusGemmLikeOp(RockGemmGemmWrapperInterface op,
 
   if (isa<AttentionOp>(op)) {
     AttentionOp attentionOp = cast<AttentionOp>(op);
-    if (failed(verifySplitKVExtraStorage(attentionOp, qBatchDim, queryM,
-                                         valueN)))
+    if (failed(
+            verifySplitKVExtraStorage(attentionOp, qBatchDim, queryM, valueN)))
       return failure();
   }