tree 98c8f24c0c3cec2f0ea7ec174bd25f3a3e9a7128
parent d58b0562a340b043ed83b0fb4375e89ed74fc621
author Jianhui Li <jian.hui.li@intel.com> 1783698706 -0700
committer GitHub <noreply@github.com> 1783698706 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJqURUSCRC1aQ7uu5UhlAAAqZ0QAD6JfgWpgF0DnhC251xvaZa7
 GBI2rmVuCPv7rwHFmh/vTUf13x3y9QAetvaYCLavvH+VZBfTB8xrSEPNP2bNfqh0
 FJUMEdXWGqC4mWMgNZr4jJp/tFkj+5jkOJUT2KzbXMUFK5Z5dFYu1uVPi9JTA5tE
 4Bi/cCO5Sboq0iKYHdXux50K3PowzLVoPJLAh91iVLAdc4iNiw2BS4Blc5wMYV2k
 1QZ9zjbEzJQ+KDzY+ycOqlD31NbdClsO6rlGMc+hbEPoCtjN0sECI9tIzB2OCQku
 xB8WcjiFQtHbdGkGL3OSm5U2PHGsiROxfuv/mBZnNVSqfxkQrq3zAi9V2MSREPt4
 fKxDsWAh+qSYGBoAW6UWmsZmIjcPUVVaWEBZkMleeAF7Q7uS0bYsK1QE/aCdl0VI
 3byO1XTzsVvtXmw/paI0pWMiYWu8142pIq5EHqq7nz4aS/b9Ux5rExFnMvmIuCt9
 Chij/B2uHGw7Pl5nKflbtnTYc9dZzr57zfFEeccSRMkeBGlT5NX2T8SkFftgrDnC
 6rqi+J9bTjcE2oIDeAeYjGENFUv5x/tE1zRn9OTowq3u7zyU29UilBsl9oMdhu0F
 hj6hEpYhCszuNJ6hsW5qbnRR+06PDOCRYMPy+pnbOH6/gwuoEGZPu7P7SSxIx1FI
 ahJ3mYGElOMzriCd1Oa4
 =q4wu
 -----END PGP SIGNATURE-----
 

[MLIR][Vector] Relax shape_cast unrolling to per-reassociation-group contiguity (#205684)

UnrollShapeCastPattern previously only unrolled a vector.shape_cast when
the target unroll tile was contiguous in the whole result vector
(isContiguous(targetShape, resultShape)). This rejected valid cases such
as:
```mlir
  %0 = vector.shape_cast %src : vector<8x32xf8> to vector<8x1x32xf8>  // target tile [8, 1, 4]
  %1 = vector.shape_cast %src : vector<8x32x32xf8> to vector<256x32xf8>  // target tile [16, 4]
```
This PR factors the source and result shapes of a shape_cast into
independent reassociation groups — maximal aligned source/result dim
ranges that hold equal element counts:
```
  - 8x32 → 8x1x32 ⇒ groups {8 ↔ 8x1}, {32 ↔ 32}
  - 8x32x32 → 256x32 ⇒ groups {8x32 ↔ 256}, {32 ↔ 32}
```
Instead of requiring the target tile to be contiguous in the whole
result vector, it only checks that the tile is contiguous within each
group.
This criterion is correct because a shape_cast preserves row-major
linear order. The groups partition the dimensions into blocks whose
element ranges don't overlap, so the global linear index breaks down
into one independent sub-index per group. A tile that is contiguous
within each group therefore corresponds to a slice that is contiguous
within each group of both the source and result.
  
     Assist-by-Claude

---------

Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>