pytorch Architecture Docs
Home / Function/ name() — pytorch Function Reference

name() — pytorch Function Reference

Architecture documentation for the name() function in generate_kernels.py from the pytorch codebase.

Function python ComputeKernels QuantizedCPU calls 1 called by 1

Entity Profile

Dependency Diagram

graph TD
  d8d80e47_3bf7_91f3_187b_0bebaae577c7["name()"]
  43bf4dfa_33b6_275e_b989_8f50c5c2b9cd["name()"]
  43bf4dfa_33b6_275e_b989_8f50c5c2b9cd -->|calls| d8d80e47_3bf7_91f3_187b_0bebaae577c7
  43bf4dfa_33b6_275e_b989_8f50c5c2b9cd["name()"]
  d8d80e47_3bf7_91f3_187b_0bebaae577c7 -->|calls| 43bf4dfa_33b6_275e_b989_8f50c5c2b9cd
  style d8d80e47_3bf7_91f3_187b_0bebaae577c7 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 82–84

    def name(self) -> str:
        acc = "rf" if self.max_k <= self.k else "gmem"
        return f"fmha_cutlassF_{self.dtype}_{self._aligned_suffix}_{self.q}x{self.k}_{acc}_sm{self.sm_range[0]}"

Domain

Subdomains

Calls

Called By

Source

Frequently Asked Questions

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

Analyze Your Own Codebase

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

Try Supermodel Free