kernels-community
/

punica-sgmv

Model card Files Files and versions Community

punica-sgmv / bgmv /bgmv_all.cu

danieldk's picture

danieldk HF Staff

Add Punica sgmv kernels

5a84343 5 days ago

history blame contribute delete

153 Bytes

	#include "bgmv_config.h"
	#include "bgmv_impl.cuh"

	FOR_BGMV_WIDE_NARROW(INST_BGMV_TWOSIDE, nv_half)
	FOR_BGMV_WIDE_NARROW(INST_BGMV_TWOSIDE, nv_bfloat16)