Replace torch.jit.script with torch.compile (#7835) #7840
tohtana
merged
10 commits into
deepspeedai:master
from
tohtana:tohtana/fix-issue7835-torch-jit-script-deprecation
Suppress import-time torch.jit deprecation warnings (#7835)
55f5c3bc
Merge branch 'master' into tohtana/fix-issue7835-torch-jit-script-dep…
90a63323
Prefer torch.compile in jit_script_compat when available (#7835)
d373def1
Keep jit_script_compat on torch.jit.script path (#7835)
25824e9f
Revert "Keep jit_script_compat on torch.jit.script path (#7835)"
959132a4
Merge branch 'tohtana/fix-issue7835-torch-jit-script-deprecation' of …
a7079851
tohtana
changed the title
Deduplicate warning handling in jit_script_compat (#7835)
9aba44d2
Simplify jit_script_compat to direct compile replacement (#7835)
a1ece2d4
Remove Muon import warning suppression block (#7835)
0b62d599
Merge branch 'master' into tohtana/fix-issue7835-torch-jit-script-dep…
a075e567
hwchen2017
approved these changes
on 2026-02-12
Login to write a write a comment.
Login via GitHub