feat: Integrate FA4 with custom BlockMask construction#96

yubofredwang merged 1 commit into

mainlightseekorg/TorchSpec:mainfrom

feat/flex-attention-fa4-blockmasklightseekorg/TorchSpec:feat/flex-attention-fa4-blockmaskCopy head branch name to clipboard