DeepSpeed
Check for local CUDA graphs when enable_cuda_graph=True
#2941
Merged

Check for local CUDA graphs when enable_cuda_graph=True #2941

jeffra merged 4 commits into master from lekurile/support_sd_cg
lekurile
lekurile Check for local CUDA graphs when enable_cuda_graph=True
dfa731af
lekurile Add CUDAGraph feature class for SD module implementations
8ff6b77f
lekurile Add __init__.py
58e7b063
lekurile lekurile marked this pull request as ready for review 2 years ago
lekurile lekurile requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 2 years ago
lekurile lekurile requested a review from jeffra jeffra 2 years ago
lekurile lekurile requested a review from mrwyattii mrwyattii 2 years ago
lekurile lekurile requested a review from awan-10 awan-10 2 years ago
lekurile lekurile requested a review from cmikeh2 cmikeh2 2 years ago
lekurile lekurile requested a review from arashb arashb 2 years ago
lekurile Merge branch 'master' into lekurile/support_sd_cg
ef815d72
jeffra
jeffra approved these changes on 2023-03-07
cmikeh2
cmikeh2 approved these changes on 2023-03-07
jeffra jeffra merged 87eaf8f9 into master 2 years ago
jeffra jeffra deleted the lekurile/support_sd_cg branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone