Add per-sequence causal policy to packed THD attention#3274

desh2608 wants to merge 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

desh2608:desh/mixed-thd-pr-minimaldesh2608/TransformerEngine:desh/mixed-thd-pr-minimalCopy head branch name to clipboard