Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[MLIR][XeGPU] Adding XeGPU 2d block operators #85804

Merged
merged 23 commits into from
Mar 20, 2024
Merged

[MLIR][XeGPU] Adding XeGPU 2d block operators #85804

merged 23 commits into from
Mar 20, 2024

Conversation

chencha3
Copy link
Contributor

This PR adds XeGPU 2D block operators. It contains:

  1. TensorDescType and TensorDescAttr definitions
  2. MemoryScopeAttr and CacheHintAttr definitions which are used by TensorDescAttr.
  3. CreateNdDescOp, PrefetchNdOp, LoadNdOp, and StoreNdOp definitions, and their corresponding testcases for illustration.

It cherry-picks daebe5c with asan fix.

chencha3 and others added 23 commits March 8, 2024 22:58
@chencha3
add TensorDescType, TensorDescAttr, and createNdDescOp.
c93bdcf
@chencha3
add prefetch_nd, load_nd, and store_nd
facb3b4
@chencha3
add test cases
9ea71f8
@chencha3
fix printformat issue and update testcases
fdd2253
@chencha3
add XeGPU 2D block operators
ad27a81
@chencha3
run clang-format
74bd038
@chencha3
sync for OffsetSizeAndStrideOpInterface
778d4d2
@chencha3
sync
3c37828
@chencha3
clean up code
b40a514
@chencha3
fix typos and improve CreateNdDescOp::verifier
b050207
@chencha3
code format
2ca12a7
@chencha3
code format
9039b5f
@chencha3
sync viewlikeOpInterface and some updates
632637e
@chencha3
fix a typo
447d623
@chencha3
code format
37a348d
@chencha3
add ViewLikeOpInterface and OffsetSizeAndStrideOpInterface to createN…
ff33828 
…dDescOp
@chencha3
fix include format
8a9df4b
@chencha3 @joker-eph
Update mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
e3857bb 
Co-authored-by: Mehdi Amini <joker.eph@gmail.com>
@chencha3
move examaple to description
5b6ebf8
@chencha3
add prop-dict
bd28ee3
@chencha3
fix asan bug
08d076b
@chencha3
run clang-format
d47a5ea
@chencha3
merge with main
7dc36b3
@chencha3 chencha3 changed the title Xegpu 2d block ops [MLIR][XeGPU] Adding XeGPU 2d block operators Mar 19, 2024
@chencha3 chencha3 mentioned this pull request Mar 19, 2024
Merged
@mshahneo mshahneo merged commit 61b24c6 into llvm:main Mar 20, 2024
5 checks passed
chencha3 added a commit to chencha3/llvm-project that referenced this pull request Mar 23, 2024
@chencha3 @joker-eph
[MLIR][XeGPU] Adding XeGPU 2d block operators (llvm#85804)
44d828b 
pickup llvm#84692 with asan fix.

---------

Co-authored-by: Mehdi Amini <joker.eph@gmail.com>
@chencha3 chencha3 deleted the xegpu_2d_block_ops branch July 25, 2024 22:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@chencha3 @mshahneo