aı
AI Compiler
테마
범위 & 출처
전체화면
한 줄의 컴파일 과정
수식 하나가 단계마다 바뀌는 모습
Python source
y =
relu
(linear(X))
NVIDIA GPU 대표 경로
외부 GEMM + 생성 pointwise 커널
발표 시작 →
INPUT
SUBTASKS
처리 순서
TRANSFORMATION
현재 표현
OUTPUT
닫기
범위 & 출처
경로: PyTorch → Inductor → Triton → NVIDIA GPU (교육용 대표 경로)
모든 IR·코드: 교육용 축약, 인수·타입 일부 생략
실제 최적화 결과: 버전·장치·shape·옵션마다 상이
FakeTensor: Dynamo가 생성, AOTAutograd가 재사용, 이후 단계는 meta["val"] 참조
ATen 변환 + decomposition: AOTAutograd 추적 중 동시 적용, linear → permute + addmm
addmm → mm + add: GPU post-grad pass, CPU는 addmm 유지
GEMM: 외부 라이브러리 호출, add + relu: 생성 커널
생략: backward, graph break
← → / Space: 단계 이동
Home·Esc: 전체 지도 / End: 마지막 결과
F: 전체화면
슬라이드 실행에 JavaScript 필요