Function f

recipes_source/debug_mode_tutorial.py:115–116

(x)

# Pre/post hash annotations report buffer hashes around each kernel call, which
# is helpful when isolating incorrect kernels.
def f(x):
    return torch.mm(torch.relu(x), x.T)

x = torch.randn(3, 3, device="cuda")