분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
SDPA rank-3 입력이 융합 커널을 건너뛰던 함정 — PyTorch 2.14에서 수정됐습니다 — DAK
배치 차원이 없거나 이미 펼쳐진 rank-3 텐서를 Scaled Dot-Product Attention에 넣으면, 예전에는 융합 CUDA 백엔드 대신 math 경로로 떨어질 수 있었습니다. PyTorch 2.14는 rank-3 입력도 융합 백엔드로 보냅니다.
자세한 내용과 근거는 DAKER 원문에서 확인하실 수 있습니다.
핵심 요약
• rank-3 SDPA 입력이 예전에는 융합 커널 대신 math 경로로 떨어질 수 있었습니다.
• PyTorch 2.14는 rank-3 입력도 융합 백엔드로 보냅니다.
• 공식 2.14 릴리스 기준으로 정리한 안내입니다.
원문 보기: https://daker.ai/community/sdpa-rank3-fusion-kernel-skip-bug-pytorch-2-14
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Co.,Ltd All rights reserved