xref: /aosp_15_r20/external/pytorch/test/bottleneck_test/test_args.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1# Owner(s): ["module: unknown"]
2
3import argparse
4
5import torch
6
7
8if __name__ == "__main__":
9    parser = argparse.ArgumentParser()
10
11    # Required args. Raises error if they aren't passed.
12    parser.add_argument("--foo", help="foo", required=True)
13    parser.add_argument("--bar", help="bar", required=True)
14    _ = parser.parse_args()
15
16    x = torch.ones((3, 3), requires_grad=True)
17    (3 * x).sum().backward()
18