main() — pytorch Function Reference

Architecture documentation for the main() function in check_graph_breaks.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking calls 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ main() — pytorch Function Reference

Dependency Diagram

graph TD
  d15ab6c5_c213_2c26_3c09_0d3b77618de4["main()"]
  c2499662_63b5_de3b_96c4_b5056c5f7e2b["check_graph_breaks()"]
  d15ab6c5_c213_2c26_3c09_0d3b77618de4 -->|calls| c2499662_63b5_de3b_96c4_b5056c5f7e2b
  style d15ab6c5_c213_2c26_3c09_0d3b77618de4 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/check_graph_breaks.py lines 120–132

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--actual", type=str, required=True)
    parser.add_argument("--expected", type=str, required=True)
    args = parser.parse_args()

    actual = pd.read_csv(args.actual)
    expected = pd.read_csv(args.expected)

    failed, msg = check_graph_breaks(actual, expected, args.expected)
    if failed:
        print(msg)
        sys.exit(1)