shapes() — pytorch Function Reference

Architecture documentation for the shapes() function in inductor_bmm.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration calls 1

Entity Profile

DynamoBenchmarks→ Configuration→ shapes() — pytorch Function Reference

Dependency Diagram

graph TD
  f8ffe77e_f49e_f15b_2b69_37243da200a0["shapes()"]
  3c19a66e_66bb_a804_2d9f_5eed5907c0cc["test_total_time()"]
  f8ffe77e_f49e_f15b_2b69_37243da200a0 -->|calls| 3c19a66e_66bb_a804_2d9f_5eed5907c0cc
  style f8ffe77e_f49e_f15b_2b69_37243da200a0 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/inductor_bmm.py lines 49–61

    shapes = [
        # BERT (all)
        ([192, 128, 64], [192, 64, 128]),
        ([192, 128, 128], [192, 128, 64]),
        # hf_GPT2 (all)
        ([12, 1024, 1024], [12, 1024, 64]),
        ([12, 1024, 64], [12, 64, 1024]),
        # hf_Albert (all)
        ([12, 512, 64], [12, 64, 512]),
        ([12, 512, 512], [12, 512, 64]),
    ]

    test_total_time(shapes)