NVIDIA는 Vulkan 1.3.300에서 VK_NV_cooperative_matrix 확장을 도입하여 VK_KHR_cooperative_matrix에서 설정된 협동 행렬 기능을 강화했습니다. 이 확장은 기본 일반 행렬 곱셈(General Matrix Multiply, GEMM) 커널을 넘어서는 기능을 가속화하는 것을 목표로 하며, 타입 및 사용 변환, 축소, 요소별 연산, 텐서 주소 지정 지원을 포함합니다.
새로운 기능은 더 유연한 행렬 크기와 작업 그룹 범위 행렬을 허용하여 사용성과 성능을 개선하는 데 중점을 두고 있으며, 이는 공유 메모리를 통한 컴파일러 관리 스테이징을 활용합니다. 이 기능은 SPV_NV_tensor_addressing 및 SPV_NV_cooperative_matrix2 SPIR-V 확장에 의해 정의되며, GLSL_NV_cooperative_matrix2 GLSL 확장과 함께 사용할 수 있습니다. 또한, 이 확장은 구현이 지원할 수 있는 행렬 유형과 차원을 나열하고, 사용 가능한 특정 기능을 지원합니다.
이번 업데이트는 Vulkan API에서 상당한 발전을 의미하며, 특히 그래픽 및 컴퓨팅 애플리케이션에서 복잡한 행렬 작업을 수행하는 개발자들에게 도움이 될 수 있으며, 이러한 새로운 기능을 활용하는 다양한 애플리케이션에서 성능 향상으로 이어질 가능성이 있습니다.
* 이 글은
phoronix.com의 기사를 요약한 것입니다. 전체 기사의 내용은
이곳에서 확인하실 수 있습니다.