Skip to content

[TTIG_PrefetchOp] Add mask argument#4011

Merged
whitneywhtsang merged 3 commits intomainfrom
whitneywhtsang/prefetch_mask
Apr 25, 2025
Merged

[TTIG_PrefetchOp] Add mask argument#4011
whitneywhtsang merged 3 commits intomainfrom
whitneywhtsang/prefetch_mask

Conversation

@whitneywhtsang
Copy link
Copy Markdown
Contributor

This PR adds a new argument mask to the prefetch operation. It is to prepare for handling prefetching of tensor of pointers, as loads from tensor of pointers can be masked.

Note: this change comes partially from #3634.

Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang self-assigned this Apr 24, 2025
@whitneywhtsang whitneywhtsang marked this pull request as ready for review April 24, 2025 20:23
@whitneywhtsang whitneywhtsang requested review from a team, alexbaden, chengjunlu and etiotto April 24, 2025 20:24
@whitneywhtsang whitneywhtsang requested a review from etiotto April 24, 2025 22:14
Comment thread test/TritonIntelGPU/tritonintelgpu.mlir
Signed-off-by: Whitney Tsang <whitney.tsang@intel.com>
@whitneywhtsang whitneywhtsang enabled auto-merge (squash) April 25, 2025 14:42
@whitneywhtsang whitneywhtsang merged commit 53b2a51 into main Apr 25, 2025
9 checks passed
@whitneywhtsang whitneywhtsang deleted the whitneywhtsang/prefetch_mask branch April 25, 2025 15:25
david-hls pushed a commit to david-hls/intel-xpu-backend-for-triton that referenced this pull request Jun 18, 2025
Co-authored-by: Tori <vwbaker@google.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants