DAKER 학습 게시 요약입니다. torch.while_loop를 CUDA 그래프에 담아 가변 반복도 한 번의 캡처로 재생합니다

핵심

· while_loop를 CUDA while 조건부 노드로 캡처합니다.

· 가변 trip count 구간을 캡처 밖으로 밀어내지 않습니다.

· 캡처 전후 동기화·스텝 시간만 비교해 보십시오.

자세히 보기: https://daker.ai/community/post-muewk8jy-130d6a3c