feat: added smem and gmem layout selector for attn kernel#490

guocuimi merged 6 commits intomainvectorch-ai/ScaleLLM:mainfrom

sm120_tma_loadvectorch-ai/ScaleLLM:sm120_tma_loadCopy head branch name to clipboard