NVIDIA SM100 GPU용 고성능 FlashAttention 및 희소 top-k 어텐션 커널 구현
요약
MiniMax에서 NVIDIA SM100 GPU를 위한 고성능 FlashAttention 및 희소 top-k 어텐션 커널 라이브러리를 공개했습니다. 이 오픈 소스 라이브러리는 고밀도 FlashAttention과 블록 희소 top-k 어텐션을 포함하며, 두 가지 JIT 컴파일 스택을 제공합니다.
핵심 포인트
- NVIDIA SM100 GPU에 최적화된 커널 구현체입니다.
- 고밀도 FlashAttention 및 희소 top-k 어텐션 기능을 지원합니다.
- csrc와 CuTe-DSL 두 가지 JIT 컴파일 스택을 제공하여 활용도가 높습니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 X @qingq77 (검증됨)의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기0