kExpansionFactor Class — pytorch Architecture

Source Code

aten/src/ATen/native/cuda/cutlass_extensions/gemm/threadblock/dq_mma_base.h lines 55–64

template<typename WarpMma, int kExpansionFactor = 1>
CUTLASS_DEVICE void run_warp_mma(WarpMma&                           warp_mma,
                                 typename WarpMma::FragmentC&       D,
                                 typename WarpMma::FragmentA const& A,
                                 typename WarpMma::FragmentB const& B,
                                 typename WarpMma::FragmentC const& C,
                                 const int                          warp_tileB_k_offset)
{
    warp_mma(D, A, B, C);
}