add comment

charithaintc · charithaintc · commit 4dfbe7c2cabc · 2025-11-24T22:41:16.000Z
diff --git a/mlir/lib/Dialect/XeGPU/Transforms/XeGPUSubgroupDistribute.cpp b/mlir/lib/Dialect/XeGPU/Transforms/XeGPUSubgroupDistribute.cpp
@@ -175,6 +175,8 @@ static bool requireTranspose(const xegpu::LayoutAttr layout,
   return laneLayout[0] == uArch->getSubgroupSize() && laneLayout[1] == 1;
 }
 
+/// Given a sequential and distributed vector type, return the list of
+/// dimensions that are distributed.
 static SmallVector<int64_t> getDistributedDims(VectorType sequentialType,
                                                VectorType distributedType) {
   assert(sequentialType.getRank() == distributedType.getRank() &&

Original file line number	Diff line number	Diff line change
`@@ -175,6 +175,8 @@ static bool requireTranspose(const xegpu::LayoutAttr layout,`
`175`	`175`	`return laneLayout[0] == uArch->getSubgroupSize() && laneLayout[1] == 1;`
`176`	`176`	`}`
`177`	`177`
	`178`	`+/// Given a sequential and distributed vector type, return the list of`
	`179`	`+/// dimensions that are distributed.`
`178`	`180`	`static SmallVector<int64_t> getDistributedDims(VectorType sequentialType,`
`179`	`181`	`VectorType distributedType) {`
`180`	`182`	`assert(sequentialType.getRank() == distributedType.getRank() &&`