Adaptation to other data types, e.g. bf16 or fp8

thanks for your intuitive blog and code!
I am thinking develop a kernel based on your kernel, but for LLM, bf16 or fp8 are mostly used.
Can you share some hints on which places in your kernel needs to changed to make it work for other data type?
