sglang_v0.5.2/pytorch_2.8.0/test/bottleneck_test/test.py

# Owner(s): ["module: unknown"]

import torch


x = torch.ones((3, 3), requires_grad=True)
(3 * x).sum().backward()