main() — pytorch Function Reference

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

Function python DynamoBenchmarks Microbenchmarking calls 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ main() — pytorch Function Reference

Dependency Diagram

graph TD
  fbb07b79_5d92_602c_7694_3dbe255e2317["main()"]
  858864ce_33a8_9654_b5b1_91feeb2494e7["check_accuracy()"]
  fbb07b79_5d92_602c_7694_3dbe255e2317 -->|calls| 858864ce_33a8_9654_b5b1_91feeb2494e7
  style fbb07b79_5d92_602c_7694_3dbe255e2317 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/check_accuracy.py lines 116–128

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_accuracy(actual, expected, args.expected)
    if failed:
        print(msg)
        sys.exit(1)