cpp_impl() — pytorch Function Reference

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

Function python ComputeKernels QuantizedCPU calls 1 called by 1

Entity Profile

ComputeKernels→ QuantizedCPU→ cpp_impl() — pytorch Function Reference

Dependency Diagram

graph TD
  cf8c3f88_b084_0f1e_7da0_062b1278122e["cpp_impl()"]
  d70d5462_0665_23fb_50f5_9d7601927e86["cpp_impl()"]
  d70d5462_0665_23fb_50f5_9d7601927e86 -->|calls| cf8c3f88_b084_0f1e_7da0_062b1278122e
  d70d5462_0665_23fb_50f5_9d7601927e86["cpp_impl()"]
  cf8c3f88_b084_0f1e_7da0_062b1278122e -->|calls| d70d5462_0665_23fb_50f5_9d7601927e86
  style cf8c3f88_b084_0f1e_7da0_062b1278122e fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/generate_kernels.py lines 108–114

    def cpp_impl(self) -> str:
        return KERNEL_IMPL_TEMPLATE.format(
            CPP_CLASS=self.cpp_class,
            NAME=self.name,
            SM=self.sm_range[0],
            SM_MAX=self.sm_range[1],
        )