Home / Function/ impl_group() — pytorch Function Reference

impl_group() — pytorch Function Reference

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

Entity Profile

Dependency Diagram

graph TD
  64aa982a_0123_4197_c44d_853e324346a3["impl_group()"]
  da9f1d69_6a83_0980_5c2d_53bb3275c796["impl_group()"]
  da9f1d69_6a83_0980_5c2d_53bb3275c796 -->|calls| 64aa982a_0123_4197_c44d_853e324346a3
  da9f1d69_6a83_0980_5c2d_53bb3275c796["impl_group()"]
  64aa982a_0123_4197_c44d_853e324346a3 -->|calls| da9f1d69_6a83_0980_5c2d_53bb3275c796
  style 64aa982a_0123_4197_c44d_853e324346a3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 103–105

    def impl_group(self) -> str:
        # Maps to file which will contain the implementation
        return f"{self.dtype}_{self._aligned_suffix}"

Subdomains

Calls

Called By

Frequently Asked Questions

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

Analyze Your Own Codebase

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

Try Supermodel Free