shapes() — pytorch Function Reference

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

Function python DynamoBenchmarks Configuration calls 1

Entity Profile

DynamoBenchmarks→ Configuration→ shapes() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/inductor_cpu_atomic.py lines 60–69

    shapes = [
        ([1, 4096]),
        ([1, 65536]),
    ]
    types = [
        torch.float32,
        torch.int32,
    ]
    print("test total time")
    test_total_time(shapes, types)