tree 63ce31e431a14fb1e64eb28f56cac9cd84eee5c5
parent 6dcc6a1a697880bc433d373938e0a8fd9d1a0f4d
author Alan Li <me@alanli.org> 1753284161 -0700
committer Copybara-Service <copybara-worker@google.com> 1753284365 -0700

[AMDGPU] fold `memref.subview/expand_shape/collapse_shape` into `amdgpu.gather_to_lds` (#149851)

This PR adds a new optimization pass to fold
`memref.subview/expand_shape/collapse_shape` ops into consumer
`amdgpu.gather_to_lds` operations.

* Implements a new pass `AmdgpuFoldMemRefOpsPass` with pattern
`FoldMemRefOpsIntoGatherToLDSOp`
* Adds corresponding folding tests

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
GitOrigin-RevId: dbc63f1e3724b6f2348c431dc1216537d9c042e8