Fix Device Segfault#152

micmelesse merged 4 commits intomain_perfROCm/flash-attention:main_perffrom

micmelesse/varlen_bwd_segfaultROCm/flash-attention:micmelesse/varlen_bwd_segfaultCopy head branch name to clipboard