name() — pytorch Function Reference
Architecture documentation for the name() function in generate_kernels.py from the pytorch codebase.
Entity Profile
Dependency Diagram
graph TD 3ab916eb_5534_c987_39b8_ae626e55c9ce["name()"] 4c399b91_7e26_1b00_534c_5c262c55c8c9["name()"] 4c399b91_7e26_1b00_534c_5c262c55c8c9 -->|calls| 3ab916eb_5534_c987_39b8_ae626e55c9ce 4c399b91_7e26_1b00_534c_5c262c55c8c9["name()"] 3ab916eb_5534_c987_39b8_ae626e55c9ce -->|calls| 4c399b91_7e26_1b00_534c_5c262c55c8c9 style 3ab916eb_5534_c987_39b8_ae626e55c9ce 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