Error in @torch.jit.script

jit
1

I am trying to learn more about the JIT compiler and was implementing the examples from the documentation, particularly this one (from TorchScript — PyTorch 2.1 documentation):

import torch
@torch.jit.script
def foo(x, y):
    if x.max() > y.max():
        r = x
    else:
        r = y
    return r

The execution of the code resulted in this error:

image
image963×451 12.4 KB

Do you guys have any ideas on what is wrong?

2

We made a change to make the bool casting more strict, try: if bool(x.max() > y.max()): instead. We need to update the docs to this effect, I’ll file a GH issue.

3
1 Like