GB7694

Graph-Break Type

Short name describing what triggered the graph break

torch.switch: unsupported branch return type (constant)

Values or code snippet captured at the break point

str(ret_val)

Explanation of why the graph break was triggered

Constants returned from branches must be int or None.

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").