본문으로 건너뛰기

vregs

벡터 레지스터 (VREGs)

중급

VREGs는 TPU의 가장 빠른 온칩 레지스터 계층으로, 벡터 연산 유닛과 MXU에 직접 데이터를 공급한다. VREGs는 레지스터 폭이 넓어 대규모 벡터/행렬 블록을 효율적으로 처리할 수 있으며, 레지스터 사용량과 레지스터 전송 타이밍이 파이프라이닝 성능을 결정한다. 커널은 VREGs에 데이터를 적시에 채워 계산 유닛의 유휴 시간을 최소화하도록 설계해야 한다.