GB0310

Graph-Break Type

Short name describing what triggered the graph break

torch.cond: improper predicate

Values or code snippet captured at the break point

str(pred)

Explanation of why the graph break was triggered

Expected pred to be a bool or a boolean tensor with a single item but got {str(type(pred))} with original python type {str(pred.python_type())}.

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