[Mlir-commits] [mlir] [MLIR][XeGPU] Support partial subgroup lane distribution (PR #201667)

[Artem Kroviakov]

================
@@ -1224,6 +1224,13 @@ struct SgToLaneVectorExtractStridedSlice
         return rewriter.notifyMatchFailure(
             op, "source of extract_strided_slice lacks distribution layout");
       int sourceDistrDimSize = op.getSourceVectorType().getShape()[distDim];
+      auto laneLayout = sourceLayout.getEffectiveLaneLayoutAsInt();
+      // Effective subgroup size needs to be adjusted if laneLayout along
+      // the distributed dimension is smaller than subgroup size.
+      if (laneLayout[distDim] < subgroupSize) {
----------------
akroviakov wrote:

Is there a reason not to unify both if conditions?

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