Make changes to non-paged attention path#3

sudhakarsingh27 wants to merge 4 commits into

cyanguwa:paged_attentioncyanguwa/TransformerEngine:paged_attentionfrom

sudhakarsingh27:pr-1355-updatessudhakarsingh27/TransformerEngine:pr-1355-updatesCopy head branch name to clipboard