export_nativert() — pytorch Function Reference

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

Function python DynamoBenchmarks Microbenchmarking calls 3 called by 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ export_nativert() — pytorch Function Reference

Dependency Diagram

graph TD
  529640b9_a20a_d7f3_29f8_900581eb0f3d["export_nativert()"]
  04a3a4a6_8db3_854d_a893_02c9542bf9dd["speedup_experiment()"]
  04a3a4a6_8db3_854d_a893_02c9542bf9dd -->|calls| 529640b9_a20a_d7f3_29f8_900581eb0f3d
  1eff8423_1f23_d138_6815_07d8dc29a749["_normalize_bench_inputs()"]
  529640b9_a20a_d7f3_29f8_900581eb0f3d -->|calls| 1eff8423_1f23_d138_6815_07d8dc29a749
  c9be2096_e6d7_2374_ad2e_a6e33f435ada["run()"]
  529640b9_a20a_d7f3_29f8_900581eb0f3d -->|calls| c9be2096_e6d7_2374_ad2e_a6e33f435ada
  3df8de63_d71a_714c_93a1_1a0e24c0f362["load()"]
  529640b9_a20a_d7f3_29f8_900581eb0f3d -->|calls| 3df8de63_d71a_714c_93a1_1a0e24c0f362
  style 529640b9_a20a_d7f3_29f8_900581eb0f3d fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 1548–1555

def export_nativert(model, example_inputs):
    optimized = NativeRTCache.load(model, example_inputs)

    def opt_nativert(_, example_inputs, collect_outputs=False):
        example_args, example_kwargs = _normalize_bench_inputs(example_inputs)
        return optimized.run(*example_args, **example_kwargs)

    return opt_nativert