in maga_transformer/cpp/cutlass/cutlass_kernels/weightOnlyBatchedGemv/kernelDispatcher.h [80:121]
void check_pointer(Params& params, cudaStream_t s)
{
if constexpr (GroupSize == 0)
{
check_alpha<Details, GroupSize, false, false, false>(params, s);
}
else
{
if (params.act_scale && params.zeros && params.bias)
{
check_alpha<Details, GroupSize, true, true, true>(params, s);
}
else if (params.act_scale && params.zeros && !params.bias)
{
check_alpha<Details, GroupSize, true, true, false>(params, s);
}
else if (params.act_scale && !params.zeros && params.bias)
{
check_alpha<Details, GroupSize, true, false, true>(params, s);
}
else if (!params.act_scale && params.zeros && params.bias)
{
check_alpha<Details, GroupSize, false, true, true>(params, s);
}
else if (!params.act_scale && !params.zeros && params.bias)
{
check_alpha<Details, GroupSize, false, false, true>(params, s);
}
else if (params.act_scale && !params.zeros && !params.bias)
{
check_alpha<Details, GroupSize, true, false, false>(params, s);
}
else if (!params.act_scale && params.zeros && !params.bias)
{
check_alpha<Details, GroupSize, false, true, false>(params, s);
}
else
{
check_alpha<Details, GroupSize, false, false, false>(params, s);
}
}
}