Add the causal mask fast path in flash attention#99

Connor1996 merged 1 commit intoskyzh:mainskyzh/tiny-llm:mainfrom

Connor1996:implConnor1996/tiny-llm:implCopy head branch name to clipboard