Add layernorm op for inputs of mixed dtype#1926

yzh119 merged 2 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

akhilg-nv:akhilg_add_layernormakhilg-nv/flashinfer:akhilg_add_layernormCopy head branch name to clipboard