fix format

Author: chencha3

mlir/test/lib/Dialect/XeGPU/TestXeGPUTransforms.cpp

@@ -44,37 +44,38 @@ struct TestXeGPUUnrollingPatterns
 
   void runOnOperation() override {
     vector::UnrollVectorOptions options;
-    options.setNativeShapeFn([&](Operation *op)
-                                 -> std::optional<SmallVector<int64_t>> {
-      if (isa<xegpu::CreateNdDescOp, xegpu::UpdateNdOffsetOp,
-              xegpu::PrefetchNdOp, xegpu::LoadNdOp, xegpu::StoreNdOp>(op)) {
-        xegpu::TensorDescType tdescTy;
-        if (auto createNdOp = dyn_cast<xegpu::CreateNdDescOp>(op)) {
-          tdescTy = createNdOp.getType();
-        } else if (auto updateNdOp = dyn_cast<xegpu::UpdateNdOffsetOp>(op)) {
-          tdescTy = updateNdOp.getTensorDescType();
-        } else if (auto prefetchNdOp = dyn_cast<xegpu::PrefetchNdOp>(op)) {
-          tdescTy = prefetchNdOp.getTensorDescType();
-        } else if (auto loadNdOp = dyn_cast<xegpu::LoadNdOp>(op)) {
-          tdescTy = loadNdOp.getTensorDescType();
-        } else if (auto storeNdOp = dyn_cast<xegpu::StoreNdOp>(op)) {
-          tdescTy = storeNdOp.getTensorDescType();
-        }
-
-        if (auto layout = tdescTy.getLayoutAttr()) {
-          auto inst_data = layout.getInstData();
-          if (inst_data && layout.isSgLayout())
-            return SmallVector<int64_t>(inst_data.asArrayRef().begin(),
-                                        inst_data.asArrayRef().end());
-        }
-      }
-
-      if (isa<xegpu::DpasOp>(op)) {
-        return SmallVector<int64_t>{8, 16, 16};
-      }
-
-      return std::nullopt;
-    });
+    options.setNativeShapeFn(
+        [&](Operation *op) -> std::optional<SmallVector<int64_t>> {
+          if (isa<xegpu::CreateNdDescOp, xegpu::UpdateNdOffsetOp,
+                  xegpu::PrefetchNdOp, xegpu::LoadNdOp, xegpu::StoreNdOp>(op)) {
+            xegpu::TensorDescType tdescTy;
+            if (auto createNdOp = dyn_cast<xegpu::CreateNdDescOp>(op)) {
+              tdescTy = createNdOp.getType();
+            } else if (auto updateNdOp =
+                           dyn_cast<xegpu::UpdateNdOffsetOp>(op)) {
+              tdescTy = updateNdOp.getTensorDescType();
+            } else if (auto prefetchNdOp = dyn_cast<xegpu::PrefetchNdOp>(op)) {
+              tdescTy = prefetchNdOp.getTensorDescType();
+            } else if (auto loadNdOp = dyn_cast<xegpu::LoadNdOp>(op)) {
+              tdescTy = loadNdOp.getTensorDescType();
+            } else if (auto storeNdOp = dyn_cast<xegpu::StoreNdOp>(op)) {
+              tdescTy = storeNdOp.getTensorDescType();
+            }
+
+            if (auto layout = tdescTy.getLayoutAttr()) {
+              auto inst_data = layout.getInstData();
+              if (inst_data && layout.isSgLayout())
+                return SmallVector<int64_t>(inst_data.asArrayRef().begin(),
+                                            inst_data.asArrayRef().end());
+            }
+          }
+
+          if (isa<xegpu::DpasOp>(op)) {
+            return SmallVector<int64_t>{8, 16, 16};
+          }
+
+          return std::nullopt;
+        });
 
     MLIRContext *ctx = &getContext();
     RewritePatternSet patterns(ctx);