trunk/33797020d6c2787535ebd2f0785b1f4ffbff37c7: Fix Dynamo tracing of torch.random.fork_rng (#186168)
- PyTorch 近 90 天出现 1435 次
- PyTorch 近 90 天第 1417 次发版
- 上一次:同一天稍早 · viable/strict/1790184407: Support tensor torch._assert in export (#186306)
发生了什么
Dynamo was classifying torch.random.fork_rng as an in-graph torch function. That is not a valid lowering target: fork_rng is a Python context manager, and proxying the call made FX try to convert its Python arguments into graph arguments. When the devices argument was a set, FX failed with NotImplementedError before Dynamo could trace the context manager body. Remove fork_rng from the non-C binding in-graph function…
摘要按规则整理自下方来源原文