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

Rewrite ReduceOp to support arbitrary reduce operations #1305

Merged
merged 23 commits into from
Apr 13, 2023
Merged

Rewrite ReduceOp to support arbitrary reduce operations #1305

Changes from 1 commit
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
5f6e3de
Add GenericReduceOp to ttir and add `tl.prod` using it
peterbell10 Mar 7, 2023
786433e
Lower tt.generic_reduce to LLVM IR
peterbell10 Mar 8, 2023
ab461e8
Support simultaneous reduction of multiple tensors
peterbell10 Mar 13, 2023
0aba718
Automatically build reduction combine op region from JITFunction
peterbell10 Mar 14, 2023
08c2e35
Replace old ReduceOp entirely
peterbell10 Mar 14, 2023
4b74ce3
Misc cleanup
peterbell10 Mar 14, 2023
4b2b16a
Add SameOperandsEncoding
peterbell10 Mar 16, 2023
b3957a7
Run clang-format
peterbell10 Mar 16, 2023
bbec2fe
Fix lit tests
peterbell10 Mar 20, 2023
7e195a6
Update to newer LLVM
peterbell10 Mar 20, 2023
0f7a528
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Mar 23, 2023
19d490b
Lint
peterbell10 Mar 23, 2023
c5d928b
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Mar 30, 2023
c6f777b
Fix merge conflicts
peterbell10 Mar 30, 2023
440a39d
Respond to some review comments
peterbell10 Apr 4, 2023
3db5241
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Apr 4, 2023
c404afe
Merge remote-tracking branch 'upstream/main' into HEAD
peterbell10 Apr 7, 2023
19d31c6
Don't rematerialize ReduceOp
peterbell10 Apr 10, 2023
a6ae9e7
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Apr 10, 2023
0cd8f0f
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Apr 11, 2023
c7c8ac1
Revert "Don't rematerialize ReduceOp"
peterbell10 Apr 12, 2023
768241d
Merge remote-tracking branch 'upstream/main' into generic-reduction
peterbell10 Apr 12, 2023
04c2164
Merge branch 'main' into generic-reduction
ptillet Apr 13, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge remote-tracking branch 'upstream/main' into generic-reduction
  • Loading branch information
@peterbell10
peterbell10 committed Mar 30, 2023
commit c5d928bfbed42fe1d40f0c35642dac2b02bc4b46
10 changes: 10 additions & 0 deletions include/triton/Dialect/Triton/IR/TritonAttrDefs.td
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,16 @@ def TT_EvictionPolicyAttr : I32EnumAttr<
let cppNamespace = "::mlir::triton";
}

def TT_PaddingOptionAttr : I32EnumAttr<
"PaddingOption", "",
[
I32EnumAttrCase<"PAD_ZERO", 1, "zero">,
// We can not set the string value to "NAN" because it is a keyword in C++
I32EnumAttrCase<"PAD_NAN", 2, "nan">
]> {
let cppNamespace = "::mlir::triton";
}

// atomic
def TT_AtomicRMWAttr : I32EnumAttr<
"RMWOp", "",
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.