2.14.11개월 전
PyTorch 2.14.1 Release
핵심 내용
PyTorch 2.14.1은 MPS 및 CUDA 환경에서의 선형 대수 연산(svd, lstsq) 관련 침묵 오류와 회귀 버그를 수정한 핫픽스 릴리스입니다.
자세히 보기
침묵 오류 수정 (Silent Correctness Fixes)
- torch.linalg.lstsq: MPS에서 복소수 배치 언더디터마인드 시스템에 대해 잘못된 해를 반환하던 문제를 수정했습니다.
- torch.linalg.svd: MPS에서 랭크 결손(rank-deficient) 및 조건이 나쁜(ill-conditioned) 입력값에 대해 U가 직교하지 않거나 작은 특이값이 부정확하게 계산되던 문제를 해결했습니다.
- CUDA 13.2.2 업데이트: NVIDIA CUDA 13.2.2 바이너리로 업데이트하여 다음 두 가지 중요한 문제를 해결했습니다.
- cuBLAS: NVFP4 행렬 곱셈 시 **cublasLtMatmul()**이 텐서 전체 스케일링을 무시하던 문제 수정.
- 컴파일러: 중첩된 스레드 분기가 있는 커널에서 재수렴 실패로 인해 레지스터 값이 손상되거나 오래된 상태로 남던 문제 수정.
회귀 버그 수정 (Regression Fixes)
- torch.linalg.svd, torch.linalg.svdvals, torch.linalg.lstsq: MPS에서 8192개 이상의 요소를 가진 입력값에 대해 Metal 파이프라인 상태 오류로 실패하던 문제를 수정했습니다.
- torch.svd(out=): MPS에서 복소수 입력값 처리 시 내부 어설션(assert)이 발생하던 문제를 수정했습니다.