DeepSpeed
Enable torch version dependent compilation of record_module and iter_params
#7362
Merged
Overview
Commits
8
Changes
View On GitHub

Enable torch version dependent compilation of record_module and iter_params #7362

tohtana merged 8 commits into deepspeedai:master from deepcharm:master
deepcharm
deepcharm deepcharm requested a review from tjruwase tjruwase 103 days ago
deepcharm deepcharm requested a review from tohtana tohtana 103 days ago
tohtana
tohtana approved these changes on 2025-06-17
deepcharm
loadams
loadams approved these changes on 2025-06-20
loadams
deepcharm
deepcharm Small fix
07e3e9f0
deepcharm Revert "Small fix"
7a1e91d9
deepcharm Avoid graph break by enabling compile of record_module
b4b24c57
deepcharm deepcharm force pushed from 0a2ca7d9 to b4b24c57 94 days ago
deepcharm Added PyTorch version compatible compiling decorator
0550de75
deepcharm deepcharm changed the title Avoid graph break by enabling compile of record module Enable torch version dependent compilation of record_module and iter_params 94 days ago
deepcharm Merge branch 'master' into master
09cf5de2
tohtana Merge branch 'master' into master
bb841d2f
loadams Merge branch 'master' into master
8988ef53
loadams Merge branch 'master' into master
541bfaef
tohtana tohtana merged 8ace4da7 into master 82 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum