- oneDNN에 AMD 젠DNN 통합 추진…UXL 재단 참여·AI 연산 명령어 공동 개발

 

AMD와 인텔이 x86 CPU에서 파이토치(PyTorch) 기반 AI 애플리케이션 개발을 단순화하기 위해 협력한다. 양사는 딥러닝 연산 라이브러리인 원DNN(oneDNN)에 각사의 CPU 최적화 기술을 결합해 개발자가 AMD와 인텔 프로세서에서 공통 소프트웨어를 활용하도록 지원할 계획이다.

 

제조사별 CPU 최적화 부담 줄이는 원DNN

AI 모델을 CPU에서 실행할 때는 행렬 곱셈과 합성곱 등 반복적인 연산을 프로세서 특성에 맞춰 처리해야 한다. CPU마다 지원하는 명령어와 연산 방식이 달라 같은 파이토치 애플리케이션에서도 하드웨어별 최적화 작업이 필요하다.

 

AMD와 인텔은 x86 생태계 자문그룹의 권고에 따라 원DNN을 파이토치의 공통 CPU 가속 라이브러리로 활용한다. 원DNN은 합성곱과 행렬 곱셈, 풀링, 정규화 등 딥러닝에 사용하는 주요 연산을 CPU에서 처리하도록 최적화하는 오픈소스 라이브러리다.

 

AMD는 자체 딥러닝 최적화 기술인 젠DNN(ZenDNN)을 원DNN에 통합한다. 개발자는 공통 라이브러리를 이용하면서 AMD와 인텔 CPU에 적용된 연산 최적화 기능을 활용할 수 있다.

 

UXL 재단 참여로 오픈소스 공동 개발

AMD는 원DNN 개발을 관리하는 통합 가속 재단(UXL Foundation)의 운영위원회에도 참여한다. 젠DNN 통합과 원DNN 개발에 참여하면서 인텔과 함께 파이토치의 x86 CPU 최적화 작업을 진행한다.

 

양사는 현재 프로세서뿐 아니라 향후 출시할 x86 CPU의 AI 연산 기능도 원DNN을 통해 파이토치에 연결할 계획이다. AMD와 인텔이 각각 개발한 CPU 최적화 기술을 하나의 소프트웨어 경로에서 제공해 개발자가 하드웨어별로 별도의 소프트웨어 스택을 선택해야 하는 부담을 줄이는 방식이다.

 

AI 컴퓨트 익스텐션으로 행렬 연산 가속

AMD와 인텔은 AI 연산을 위한 새로운 x86 명령어 확장 기술인 AI 컴퓨트 익스텐션(AI Compute Extensions)도 공동 개발하고 있다. AI 모델에서 사용하는 행렬 곱셈 연산을 CPU에서 처리하는 기술로, 양사는 향후 원DNN에 해당 명령어를 반영해 파이토치가 새로운 CPU 연산 기능을 활용하도록 할 계획이다. 구체적인 성능 향상 수치와 적용 시기는 이번 발표에서 공개하지 않았다.

 

AMD·인텔, x86 AI 소프트웨어 공동 대응

인텔 데이터센터 전략 프로그램 부문 제프 맥베이(Jeff McVeigh) 부사장 겸 총괄은 “개발자가 x86 CPU 성능을 활용하기 위해 서로 다른 소프트웨어 스택 가운데 하나를 선택해야 해서는 안 된다”며 “원DNN을 중심으로 AMD와 협력해 파이토치 개발자에게 AMD와 인텔 CPU를 아우르는 일관된 개발 환경을 제공하겠다”고 밝혔다.

 

AMD 아키텍처·전략 부문 로버트 호머스(Robert Hormuth) 기업 부사장은 “AMD는 개발자에게 선택권을 제공하고 고성능 AI 애플리케이션 개발을 지원하는 개방형 소프트웨어 생태계를 발전시키는 데 집중하고 있다”며 “x86 생태계 자문그룹과 협력해 AMD와 인텔 프로세서 전반에서 개발 작업을 단순화하는 공통 기술을 발전시키겠다”고 설명했다.

 

레드햇 최고기술책임자실 스티븐 와트(Stephen Watt) 부사장 겸 수석 엔지니어는 “AMD와 인텔이 원DNN을 중심으로 파이토치의 x86 CPU 최적화 작업을 통합하는 것은 개발자가 서로 다른 x86 프로세서에서 AI 애플리케이션을 개발하고 실행하는 과정을 단순화하는 데 도움이 될 것”이라고 평가했다.

 

#AMD #인텔 #파이토치 #PyTorch #원DNN #oneDNN #젠DNN #ZenDNN #x86 #CPU #AI개발 #AI컴퓨트익스텐션 #UXL재단