엔비디아 cuBLAS, 과학 시뮬레이션 속도 높일 알고리즘 추가

엔비디아의 CUDA 툴킷 13.4.2에 FP64 행렬 곱셈을 위한 오자키-II와 2M 알고리즘이 cuBLAS에 추가됐다. 엔비디아는 이 알고리즘이 FP64 정확도를 유지하면서 과학 연산 작업을 가속한다고 밝혔다. 엔비디아에 따르면 오자키-II DGEMM은 VR NVL72 GPU에서 기본 FP64보다 5.5배, GB200 GPU에서 3.9배 빠르며 GB200에서 오자키-I보다 2배 빠르다. 2M ZGEMM은 GB200에서 기본 복소수 FP64보다 5배 빠르다. 단일 및 이중 여기 결합 클러스터를 사용한 양자화학 시뮬레이션에서 오자키-II는 전체 처리 속도를 VR NVL72에서 2배, GB200에서 1.4배 높였다. 혼합 정밀도 모드에서는 각각 5배와 3.5배로 높아졌다. 버클리GW 재료과학 시뮬레이션에서 2M ZGEMM은 기본 ZGEMM보다 VR NVL72에서 6.6배, GB200에서 3.9배 빨랐다. 이 알고리즘은 정수 텐서 코어를 사용한다. cuBLAS의 자동 동적 정밀도 프레임워크는 수치 정확도를 유지하기 위해 필요할 때 기본 FP64 명령어로 전환할 수 있다. 엔비디아는 TI16 형식과 tcgen05.mma 명령어가 루빈 GPU에서 오자키-I과 오자키-II를 더욱 가속한다고 밝혔다. 개발자는 CUDA 툴킷 13.4.2에서 이 알고리즘을 사용할 수 있다.