Function foo

recipes_source/torch_compiler_set_stance_tutorial.py:37–43

(x)

@torch.compile
def foo(x):
    if torch.compiler.is_compiling():
        # torch.compile is active
        return x + 1
    else:
        # torch.compile is not active
        return x - 1


inp = torch.zeros(3)