[ROCm] support triton-based flash-attn in TE#177

wangye805 wants to merge 1 commit intodevROCm/TransformerEngine:devfrom

yewang12/triton-based-faROCm/TransformerEngine:yewang12/triton-based-faCopy head branch name to clipboard