get_benchmark_indices() — pytorch Function Reference

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

Function python DynamoBenchmarks Configuration called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ get_benchmark_indices() — pytorch Function Reference

Dependency Diagram

graph TD
  e9a82af3_ba81_409b_9d46_c9e701ace22a["get_benchmark_indices()"]
  a022c1fd_0072_4493_ca52_f1e989604c2c["iter_model_names()"]
  a022c1fd_0072_4493_ca52_f1e989604c2c -->|calls| e9a82af3_ba81_409b_9d46_c9e701ace22a
  style e9a82af3_ba81_409b_9d46_c9e701ace22a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 2087–2094

    def get_benchmark_indices(self, length):
        start = self._args.partition_id * (length // self._args.total_partitions)
        end = (
            (self._args.partition_id + 1) * (length // self._args.total_partitions)
            if self._args.partition_id < self._args.total_partitions - 1
            else length
        )
        return start, end