[Mlir-commits] [mlir] [mlir][XeGPU][Transform] Add gather/scatter coalescing analysis. (PR #201684)

[Charitha Saumya]

================
@@ -80,6 +80,18 @@ void populateXeGPUSgToLaneDistributeTypeConversionAndLegality(
     TypeConverter &typeConverter, RewritePatternSet &patterns,
     ConversionTarget &target);
 
+//===----------------------------------------------------------------------===//
+// Coalesce gather/scatter analysis + apply.
+//===----------------------------------------------------------------------===//
+
+/// Run the AxisInfo-based contiguity analysis over `root` and stamp a
+/// `contiguous_chunk` attribute on every `xegpu.load` / `xegpu.store` whose
+/// offsets are contiguous (in runs of >= 2) along the innermost dimension.
+/// The stamped value is the inner-dim contiguity; it is a target-independent
+/// property consumed downstream (e.g. to derive a `lane_data` split). Ops that
+/// already carry a `contiguous_chunk` attribute are left untouched.
+void runCoalesceGatherScatterAnalysis(Operation *root);
----------------
charithaintc wrote:

who is the planned user for this method? will it be run before layout prop?

https://github.com/llvm/llvm-project/pull/201684