Skip to content

custom_kernel: fix shape mismatch by sharding segment_ids in flash attn.#8333

Draft
dudulightricks wants to merge 2 commits intopytorch:masterfrom dudulightricks:bug-fix/shard-segment-ids-flash-attention
+58-3

Commits

Commits on Nov 25, 2024