get_suite_from_model_iter_fn() — pytorch Function Reference

Architecture documentation for the get_suite_from_model_iter_fn() function in common.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking called by 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ get_suite_from_model_iter_fn() — pytorch Function Reference

Dependency Diagram

graph TD
  cae7ee8e_ad93_67b5_9bdd_648c2e822459["get_suite_from_model_iter_fn()"]
  04a3a4a6_8db3_854d_a893_02c9542bf9dd["speedup_experiment()"]
  04a3a4a6_8db3_854d_a893_02c9542bf9dd -->|calls| cae7ee8e_ad93_67b5_9bdd_648c2e822459
  style cae7ee8e_ad93_67b5_9bdd_648c2e822459 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 421–428

def get_suite_from_model_iter_fn(model_iter_fn):
    # TODO: This is a bit of a hack
    suite = None
    if (runner := getattr(model_iter_fn, "__self__", None)) and hasattr(
        runner, "suite_name"
    ):
        suite = runner.suite_name
    return suite