__post_init__() — pytorch Function Reference

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

Function python ComputeKernels QuantizedCPU calls 1 called by 1

Entity Profile

ComputeKernels→ QuantizedCPU→ __post_init__() — pytorch Function Reference

Dependency Diagram

graph TD
  09ea15c6_e3f3_3717_43c9_3ba9e722f8e2["__post_init__()"]
  b6356833_87b5_3a29_f757_dbd0dd087434["__post_init__()"]
  b6356833_87b5_3a29_f757_dbd0dd087434 -->|calls| 09ea15c6_e3f3_3717_43c9_3ba9e722f8e2
  b6356833_87b5_3a29_f757_dbd0dd087434["__post_init__()"]
  09ea15c6_e3f3_3717_43c9_3ba9e722f8e2 -->|calls| b6356833_87b5_3a29_f757_dbd0dd087434
  style 09ea15c6_e3f3_3717_43c9_3ba9e722f8e2 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 62–75

    def __post_init__(self) -> None:
        # Set kernel selection priority
        # The lowest value that matches inputs
        # will be selected
        self.sort_index = (
            # First select aligned kernel
            0 if self.aligned else 1,
            # Then keep output in RF
            self.max_k,
            self.k,
            # Prefer kernels without dropout/bias if available
            1 if self.supports_dropout else 0,
            1 if self.supports_bias else 0,
        )