trunk/59356ef48b46afa2f817c09bf6b0cbd224f74314: Fix export move_to_device_pass device rewrites (#185105)
- PyTorch 近 90 天出现 1470 次
- PyTorch 近 90 天第 1451 次发版
- 上一次:同一天稍早 · trunk/377ffb934bd31a1538eb67a5ee1bc5cba10a4660: AArch64: FP16 matmul (#198498)
发生了什么
move_to_device_pass rewrites device-bearing FX nodes and tensor metadata, but it did not recompile the affected GraphModules. Generated Python could therefore retain stale _assert_tensor_metadata device arguments and fail when the moved graph ran directly through ep.graph_module . Recompile every GraphModule touched by the pass so its callable code matches the rewritten graph. Export also supports interpreter-only g…
摘要按规则整理自下方来源原文