pass params_dtype to qk_norm creation#2718

pstjohn wants to merge 3 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

pstjohn:pstjohn/qk-norm-dtypepstjohn/TransformerEngine:pstjohn/qk-norm-dtypeCopy head branch name to clipboard