github-featured / deepspeedai--deepspeed

发布

0 发布 128 Git 标签

RSS 订阅源
  • v0.19.1 510ebe58e4
    比较

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

    frostbyte_neo 发布于 2026-05-25 16:24:53 +00:00

    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

    下载附件