add SD injection policy #2381
jeffra
merged
29 commits into
master
from
jeffra/sd-policy
add initial sd policy
df895f7a
Merge branch 'master' into jeffra/sd-policy
d0576d48
formatting
c8163e28
add attention kernel and enable cuda-graph for SD models
f34eb5bb
add new files
5aec40c2
formatting and add dtype to unet
23674fd8
adding more optitmization by enabling ds-encoder with CUDA-Graph
6bf0c738
Merge branch 'master' into jeffra/sd-policy
8b0762b9
add missing file
ac29fdcc
Merge branch 'jeffra/sd-policy' of github.com:microsoft/DeepSpeed int…
c1770708
adapt the triton kernel to be used in more places
27e752be
add more fusion
a40780b5
allocate workspace using the padded hidden_size
923ddd2f
skip the clip-encoder injection for now
2dca3acd
Merge branch 'master' into jeffra/sd-policy
52dd412e
add triton to new extra
6079065f
lazy import triton, add sd extra, formatting
fb1605f3
delay import
16bcef13
fix previous issue i added
691fd349
fix bug with adding bias
c758b03e
Merge branch 'master' into jeffra/sd-policy
da013373
fixes for triton import and add acks to triton-ops file
406832c5
Merge branch 'jeffra/sd-policy' of github.com:microsoft/DeepSpeed int…
79b05ca1
Merge branch 'master' into jeffra/sd-policy
a36275c1
merge fix & formatting
75fbcfe7
Merge branch 'jeffra/sd-policy' of github.com:microsoft/DeepSpeed int…
e6175955
fix small issue
eff95e7e
skip cuda-graph for clip-encoder for now (it has issue on larger batc…
770e88b1
Merge branch 'master' into jeffra/sd-policy
31488815
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub