Home / Function/ inductor_triton_bmm() — pytorch Function Reference

inductor_triton_bmm() — pytorch Function Reference

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

Entity Profile

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/inductor_bmm.py lines 15–16

def inductor_triton_bmm(a, b):
    return torch.bmm(a, b)

Subdomains

Called By

Frequently Asked Questions

What does inductor_triton_bmm() do?
inductor_triton_bmm() is a function in the pytorch codebase.
What calls inductor_triton_bmm()?
inductor_triton_bmm() is called by 1 function(s): test_total_time.

Analyze Your Own Codebase

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

Try Supermodel Free