FragmentElementwiseScaler Class — pytorch Architecture
Architecture documentation for the FragmentElementwiseScaler class in mma_from_smem.h from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/mma_from_smem.h lines 304–304
class FragmentElementwiseScaler;
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free