GB9203

Graph-Break Type

Short name describing what triggered the graph break

torch.switch: differing branch outputs

Values or code snippet captured at the break point

branch{i}: {spec_a.treespec}, branch{i + 1}: {spec_b.treespec}, same_spec: {same_spec}

Explanation of why the graph break was triggered

Expected branches to return the same pytree structure.

Hints on how to resolve the graph break

Your code may result in an error when running in eager. Please double check that your code doesn’t contain a similar error when actually running eager/uncompiled. You can do this by removing the torch.compile call, or by using torch.compiler.set_stance("force_eager").