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

add triton_group_norm #4

Open
wants to merge 19 commits into
base: jieru_triton
Choose a base branch
from
Open

add triton_group_norm #4

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
16669e1
add triton_group_norm
nishirong Apr 12, 2024
430bc74
If the grid parameter is 1, can not be hinted
nishirong Apr 15, 2024
87be2fa
Comment out cout output and preserve tensor without clearing in gener…
nishirong Apr 22, 2024
803b1f0
Modify autotune for clearing the result
nishirong Apr 22, 2024
a03bf52
Merge remote-tracking branch 'zkk/jieru_triton' into group_norm_aot
nishirong Apr 22, 2024
f33d968
Modify hint code
nishirong Apr 22, 2024
368150b
fix something
nishirong Apr 22, 2024
9591df5
fix some bug for export static model
nishirong Apr 23, 2024
a4cb699
comment something
nishirong Apr 23, 2024
12b0b80
Merge remote-tracking branch 'zkk/jieru_triton' into group_norm_aot 2
nishirong Apr 23, 2024
fa92bac
assert H>0 and W>0
nishirong Apr 23, 2024
987d499
assert H>0 and W>0 2
nishirong Apr 23, 2024
9834e11
Set weight and bias as required item
nishirong Apr 23, 2024
4a86e8a
merge
nishirong Apr 24, 2024
925fcfd
Remove two redundant sentences for group_norm kernel
nishirong Apr 24, 2024
ff7713d
merge 2.0
nishirong Apr 26, 2024
6edca2f
Remove three redundant import
nishirong Apr 26, 2024
2836f3a
commit
nishirong Apr 28, 2024
b3e997c
comment out only support float16
nishirong May 6, 2024
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
Loading