Home / Function/ triton_mm() — pytorch Function Reference

triton_mm() — pytorch Function Reference

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

Function python DynamoBenchmarks Configuration called by 1

Entity Profile

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/inductor_mm.py lines 31–32

def triton_mm(a, b):
    return triton.ops.matmul(a, b)

Domain

Subdomains

Called By

Source

Frequently Asked Questions

What does triton_mm() do?
triton_mm() is a function in the pytorch codebase.
What calls triton_mm()?
triton_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