SFU-Sim: NVIDIA MUFU의 비트 정확 역공학 및 PTX 경로 검증

원문 발행 2026. 11. 10. 19:38원문 언어 영어AI 한국어 번역arXiv cs.AR 원문 보기

요약

SFU-Sim은 NVIDIA MUFU 명령어의 비트 정확 역공학 모델을 구축한 오픈 소스 도구입니다. 이 시뮬레이터는 단일 기본 명령부터 전체 PTX 경로까지 포괄적으로 모델링하며, Ampere, Ada Lovelace, Blackwell 등 여러 아키텍처를 비교 검증합니다. 하드웨어 비트와 완벽하게 일치하는 출력을 제공하여 명령어 시뮬레이션 및 기능 검증에 활용 가능합니다.

핵심 포인트

SFU-Sim은 NVIDIA MUFU 명령어의 비트 정확 역공학 모델을 구축했습니다.
단일 기본 명령부터 전체 PTX 경로까지 포괄적으로 다루며, SIN/COS 분석이 가능합니다.
Ampere, Ada Lovelace, Blackwell 등 3개 아키텍처에서 검증되었습니다.
하드웨어 비트와 완벽하게 일치하는 출력을 제공하며 조회 테이블 접근이 필요 없습니다.

GPU의 수치적 동작을 재현하려면 특수 함수 명령어(special function instructions)의 정확한 출력을 예측해야 합니다. 공급업체 문서에는 함수 의미론(function semantics)과 오차 범위만 제공할 뿐, 출력 비트 패턴은 고정하지 않습니다.

우리는 실제 GPU에서 블랙박스 역공학(black-box reverse engineering)을 통해 NVIDIA 다기능 유닛(multi-function unit, MUFU) 명령어의 비트 정확 모델을 구축한 SFU-Sim을 소개합니다. 먼저 단일 기본(bare) MUFU 명령어를 모델링합니다. 그런 다음 디스어셈블리에서 발견된 주변 연산을 추가하여 전체 PTX 경로를 모델링합니다. 이 모델은 테스트된 Blackwell GPU에서 12가지 일반적인 MUFU 형태와 두 가지 BF16 형태를 다룹니다.

완전한 사례 연구로 SIN/COS를 사용하여, 우리가 어떻게 PTX-to-SASS 컴파일 경로를 분석하고 세그먼트 구조(segment structure), 등가 계수 테이블(equivalent coefficient tables), 그리고 유한 폭 산술(finite-width arithmetic)을 복구하는지 보여줍니다. 세 가지 아키텍처(Ampere, Ada Lovelace, 그리고 Blackwell)의 NVIDIA GPU에서, 우리는 지정된 입력 도메인에 걸쳐 테스트된 모든 PTX 경로를 철저하게 비교합니다. FP16 및 FP32 경로는 각각 $2^{16}$ 및 $2^{32}$개의 모든 입력 인코딩을 다룹니다.

모델의 출력은 하드웨어 비트와 비트 단위로 일치하며, 불일치는 없습니다(zero mismatches). SFU-Sim은 각 출력을 입력으로부터 직접 계산합니다. 이는 모든 입력을 위한 조회 테이블(lookup table)이나 물리적 GPU에 대한 접근이 필요하지 않습니다. 따라서 명령어 시뮬레이터용 비트 정확 특수 함수 컴포넌트로 사용될 수 있습니다. 또한 동일한 수치적 동작을 목표로 하는 하드웨어 설계 및 기능 검증을 위한 산술 참조 자료로도 사용될 수 있습니다.

SFU-Sim과 그 검증 프로그램은 https://github.com/hzX40000000/SFU-Sim에서 오픈 소스로 공개되었습니다.

AI 자동 생성 콘텐츠

원문 바로가기

SFU-Sim: NVIDIA MUFU의 비트 정확 역공학 및 PTX 경로 검증

요약

핵심 포인트

댓글