torch_mm() — pytorch Function Reference

Architecture documentation for the torch_mm() function in inductor_mm.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking called by 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ torch_mm() — pytorch Function Reference

Dependency Diagram

graph TD
  3994379c_800b_4e6e_b053_5276e27e0062["torch_mm()"]
  dd08baca_e7b7_4c73_60cc_7d9f6fa55a3a["test_GPU_time()"]
  dd08baca_e7b7_4c73_60cc_7d9f6fa55a3a -->|calls| 3994379c_800b_4e6e_b053_5276e27e0062
  style 3994379c_800b_4e6e_b053_5276e27e0062 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/inductor_mm.py lines 27–28

def torch_mm(a, b):
    return torch.mm(a, b)

Domain

DynamoBenchmarks

Subdomains

Microbenchmarking

Called By

test_GPU_time()

Source

View on GitHub

Frequently Asked Questions

What does torch_mm() do?

torch_mm() is a function in the pytorch codebase.

What calls torch_mm()?

torch_mm() is called by 1 function(s): test_GPU_time.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free