MCPcopy Create free account
hub / github.com/KnowingNothing/MatmulTutorial / reference_gpu_gemm

Function reference_gpu_gemm

include/reference.h:79–90  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

77
78template<class AType, class BType, class CType, class AccumType>
79void reference_gpu_gemm(GemmParams<AType, BType, CType, AccumType> params) {
80 /// don't know what tilings are good, but not important here
81 const int mTile = 4;
82 const int nTile = 4;
83 const int kTile = 4;
84 const int tx = 128;
85 const int ty = 1;
86 dim3 grid(ceil_div(params.N, nTile * tx), ceil_div(params.M, mTile * ty));
87 dim3 block(tx, ty);
88 reference_gpu_gemm_kernel<AType, BType, CType, AccumType, mTile, nTile, kTile><<<grid, block>>>(params);
89 CUDA_CHECK(cudaGetLastError());
90}

Callers

nothing calls this directly

Calls 1

ceil_divFunction · 0.85

Tested by

no test coverage detected