feat(dflash): add flex_attention with BlockMask support#435

sleepcoo merged 1 commit intosgl-project:mainsgl-project/SpecForge:mainfrom

eigen-ai-labs:feature/dflasheigen-ai-labs/SpecForge_public:feature/dflashCopy head branch name to clipboard