[Core] add QKV fusion to AuraFlow and PixArt Sigma #8952
add fusion support to pixart
40ad9c80
add to auraflow.
48dfbb6f
add tests
2ff7a4a3
Merge branch 'main' into qkv-rest
26c687c2
Merge branch 'main' into qkv-rest
e933bde2
Merge branch 'main' into qkv-rest
c3ce92ca
yiyixuxu
approved these changes
on 2024-07-25
Merge branch 'main' into qkv-rest
7734a3dd
apply review feedback.
62f2af1b
Merge branch 'main' into qkv-rest
eca8d2a7
add back args and kwargs
33310efc
style
0e5036db
Merge branch 'main' into qkv-rest
c1e6a46c
Merge branch 'main' into qkv-rest
544391d8
Merge branch 'main' into qkv-rest
9171f8df
resolve brutal conflicts.
0e61ed28
Merge branch 'main' into qkv-rest
632e5839
Merge branch 'main' into qkv-rest
8ed33074
yiyixuxu
merged
52f1378e
into main 1 year ago
yiyixuxu
deleted the qkv-rest branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub