← 返回事件
持续讨论AI发版

trunk/33797020d6c2787535ebd2f0785b1f4ffbff37c7: Fix Dynamo tracing of torch.random.fork_rng (#186168)

发生了什么

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…

摘要按规则整理自下方来源原文

为什么在扩散

来源