name() — pytorch Function Reference

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

Function python CoreTensor Dispatcher calls 1 called by 1

Entity Profile

CoreTensor→ Dispatcher→ name() — pytorch Function Reference

Dependency Diagram

graph TD
  4c399b91_7e26_1b00_534c_5c262c55c8c9["name()"]
  3ab916eb_5534_c987_39b8_ae626e55c9ce["name()"]
  3ab916eb_5534_c987_39b8_ae626e55c9ce -->|calls| 4c399b91_7e26_1b00_534c_5c262c55c8c9
  3ab916eb_5534_c987_39b8_ae626e55c9ce["name()"]
  4c399b91_7e26_1b00_534c_5c262c55c8c9 -->|calls| 3ab916eb_5534_c987_39b8_ae626e55c9ce
  style 4c399b91_7e26_1b00_534c_5c262c55c8c9 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 182–190

    def name(self) -> str:
        dropout_suffix = "_dropout" if self.apply_dropout else ""
        seqlen_aligned_suffix = (
            "_seqaligned" if self.keys_queries_aligned_to_blocksizes else ""
        )
        return (
            f"fmha_cutlassB_{self.dtype}_{self._aligned_suffix}"
            f"_{self.block_i}x{self.block_j}_k{self.max_k}{dropout_suffix}{seqlen_aligned_suffix}_sm{self.sm_range[0]}"
        )