Haicheng Wu

Results 323 comments of Haicheng Wu

What is the name of the kernel used by cublas?

You can write your own via epilogue visitor tree (evt). We have an example of using evt.

Thanks for your interest. Actually, it is better asking this question in Pytorch github. They integrated many cutlass kernels (https://github.com/pytorch/pytorch/pulls?q=cutlass) already. cc += @jackkosaian

cutlass already supports fp8 sparse gemm. as all other sparse gemm, it only has row major output. if you want to implement col major output, first thing is that you...