DeepSpeed
510ebe58 - Fix DeepCompile AOT kwargs patching for PyTorch >= v2.11 (#8024)

Commit
15 days ago
Fix DeepCompile AOT kwargs patching for PyTorch >= v2.11 (#8024) DeepCompiles breaks for PyTorch >= v2.11 because these versions can construct the AOT Autograd backend without a bw_compiler kwarg, while DeepSpeed's Inductor patch assumes that key is always present. This PR fixes DeepCompile's AOT Autograd patch so unrelated AOT backend registrations can pass through unchanged. `TestDeepCompile` passes with this fix. Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>
Author
Parents
Loading